Invaild device specified in common.py

Competition_3v3snakes/rl_trainer/common.py Line 11
device = torch.device("cuda:1") if torch.cuda.is_available() else torch.device("cpu")

"cuda:1" should be replcaed by "cuda:0", or there is an error "CUDA error: invalid device ordinal"

Because "cuda:1" chooses the second gpu.