解决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解决pytorch错误ImportError: Failed to load PyTorch C extensions:和RuntimeError:CUDA error:unknown error_第1张图片
第二步:点击Python Interpreter解决pytorch错误ImportError: Failed to load PyTorch C extensions:和RuntimeError:CUDA error:unknown error_第2张图片
第三步:点击右上角小齿轮,然后点击add
解决pytorch错误ImportError: Failed to load PyTorch C extensions:和RuntimeError:CUDA error:unknown error_第3张图片
第四步:在打开的界面中依次点击如下步骤。
解决pytorch错误ImportError: Failed to load PyTorch C extensions:和RuntimeError:CUDA error:unknown error_第4张图片
第五步:选择正确的python编辑器。在Anaconda安装文件夹下选择“python.exe”,然后点击OK就成功了。
解决pytorch错误ImportError: Failed to load PyTorch C extensions:和RuntimeError:CUDA error:unknown error_第5张图片

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

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

你可能感兴趣的:(pytorch,python,深度学习)