解决pytorch错误ImportError: Failed to load PyTorch C extensions:和RuntimeError:CUDA error:unknown error

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打开Settings
第二步:点击Python Interpreter
第三步:点击右上角小齿轮,然后点击add

第四步:在打开的界面中依次点击如下步骤。

第五步:选择正确的python编辑器。在Anaconda安装文件夹下选择“python.exe”,然后点击OK就成功了。

2.解决错误RuntimeError:CUDA error:unknown error

在import torch下面一行添加“torch.cuda.current_device()”就可以解决了。

猜你喜欢

转载自blog.csdn.net/aixuexideou/article/details/125693484