Hi,
My code wok fine on Colabolatory with CUDA.
Now I try to do on local environment.
My env is;
CUDA: 10.1
cuDNN: 7

and today I installed PyTorch follows instruction of pip3.
After the setting up I meet an error;
RuntimeError: expected device cpu and dtype Float but got device cuda:0 and dtype Float

Acutually my code set device(“cuda”), I want it run on NVIDIA GPU.
I can not understand reason why script try to run on CPU, while Colab has no error (indeed run on cpu).
In addition, I replaced operators with torch’s operators, but still I get the error.
Could you tell me reason and its solution?