Hi, I have this problem also. But my setting is different. I have CUDA 7.5 but with Intel GPU instead. I run with Pytorch with CPU only but still not work. My error message is posted here. Thank you in advance.
Thank you very much for your kind reply. I tried to uninstall GPU version and install CPU version of Pytorch. However, the error message now is “AttributeError: module ‘torch._C’ has no attribute ‘_cuda_getDevice’”. I am wondering is it possible that the pre-trained model I am using based on GPU?