Hello everyone,
Today I installed anaconda 5.0.1 (Python 3.6) and Pytorch by command: conda install pytorch -c pytorch. It installs automatically cuda toolkit and pytorch. However, when I run my python program on GPU, I meet this error:
THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1532502421238/work/aten/src/THC/THCGeneral.cpp line=663 error=11 : invalid argument
My PC has 4 Cards NVIDIA GTX 1080 Ti. I chose device with device=torch.device(‘cuda:3’). But the problem’s still display. How can I fix this problem?