Competition_3v3snakes icon indicating copy to clipboard operation
Competition_3v3snakes copied to clipboard

Invaild device specified in common.py

Open fanqiNO1 opened this issue 3 years ago • 0 comments

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.

fanqiNO1 avatar Nov 12 '21 15:11 fanqiNO1