Getting the error 'invalid device ordinal #2
Description
On running the main.py with given code , I m getting this error
cuda runtime error (10) : invalid device ordinal at torch/csrc/cuda/Module.cpp:32
Can you please tell how to fix it , i have checked on the google but could not fix it on my ubuntu 16.04 terminal. It would be great help
Thanks !