这是我参与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,问题解决!!