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?