根本原因是cudatoolkit版本和pytorch版本和cuda版本不对应?
1. 查看cuda版本
cat /usr/local/cuda/version.txt

参考:Linux 和 Windows 查看 CUDA 和 cuDNN 版本
2. 找对应的cudatoolkit版本
参考binary-compatibility找到对应的 cudatoolkit对应的版本进行安装 。

3. 安装对应的cudatoolkit
这里的10.0要换成上文中查找到的对应的cuda版本。
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
其他
(1) AttributeError: module ‘torch’ has no attribute ‘distributed’
也是 版本不对应的问题?

本文详细介绍了如何解决PyTorch与CUDA版本不匹配导致的问题,包括查看CUDA版本、选择正确的cuDAtOOLKIT版本进行安装,以及解决AttributeError等常见错误。

2277

被折叠的 条评论
为什么被折叠?



