Competition_3v3snakes
Competition_3v3snakes copied to clipboard
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.