search

handria-ntoanina / unity-ml-tennis

About solving the tennis environment from Unity using a Multi Agent DDPG and a Multi Agent PPO
8 stars 3 forks source link

CUDA error (10): invalid device ordinal #2

Open Usmaniatech opened 5 years ago

Usmaniatech commented 5 years ago

I am using cuda to work on MADDPG and it gives following error:

CUDA error (10): invalid device ordinal

I am using:

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

device = "cpu"

to train on cuda. but i get the error.