1.解决错误ImportError: Failed to load PyTorch C extensions:
在安装torch,并且把"D:\py Anaconda\Lib\site-packages"下的torch相关文件复制到“D:\py Anaconda\envs\pychome\Lib\site-packages”后,运行pychram时错误“ImportError: Failed to load PyTorch C extensions:”。(注意,本人安装Anaconda的文件夹叫做“py Anaconda”。)
出现该错误并不代表是pytorch安装错误或者pytorch与CUDA版本不匹配,出现该错误的原因是没有选对python编辑器。解决办法:
第一步:File->Setting
第二步:点击Python Interpreter
第三步:点击右上角小齿轮,然后点击add
第四步:在打开的界面中依次点击如下步骤。
第五步:选择正确的python编辑器。在Anaconda安装文件夹下选择“python.exe”,然后点击OK就成功了。
2.解决错误RuntimeError:CUDA error:unknown error
在import torch下面一行添加“torch.cuda.current_device()”就可以解决了。