pytorch踩坑记录

这是我参与11月更文挑战的第1天

1.UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 10010)

分析:

提示错误原因是CUDA版本过低，即CUDA version < torch version，由此我们有两种解决办法：

• 升高CUDA版本
• 降低torch版本

排查:

print(torch.version.cuda)
复制代码

查看torch使用的cuda版本，结果为10.2

nvcc -V
复制代码

查看目前显卡的cuda版本发现为10.1 至此，我们可以得知错误是由显卡与torch的cuda版本不匹配导致的

解决：

点击以下链接，进入pytorch官网寻找cuda10.1支持的版本 pytorch.org/get-started… 选择torch1.7.1对应cuda10.1的版本下载安装，由于国内不支持conda，所以选择pip安装

# CUDA 10.1
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
复制代码

验证：

再次运行程序error消失，torch.version.cuda也变回10.1，问题解决！！

References：

1. blog.csdn.net/weixin_4373…
2. www.zhihu.com/question/30…