RuntimeError: CUDA error: invalid device ordinal

divelab / DIG

A library for graph deep learning research

https://diveintographs.readthedocs.io/

GNU General Public License v3.0

1.81k stars 281 forks source link

Closed langineer closed 9 months ago

langineer commented 9 months ago

Trying to run example_grace.ipynb. And keep running into this error: RuntimeError: CUDA error: invalid device ordinal

Cuda is available, and the device ID is correct. Pytorch 1.13 cuda 11.6 torch_geometric 2.3.1 dig 1.1.0

What could be the issue?

langineer commented 9 months ago

my mistake, didn't pass the correct device in the model parameters

langineer commented 9 months ago

closing