问题:遇到如下报错,
RuntimeError: The NVIDIA driver on your system is too old (found version 10010). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver.
并不是CUDA版本太老,而是pytorch版本和CUDA不一致。
查看pytorch版本
python
import torch
print(torch.__version__)
查看CUDA版本
cmd输入
nvidia-smi
如果找不到,是因为路径未加入path。
简单方法:找到C:\Program Files\NVIDIA Corporation\NVSMI路径,在此进入cmd。
我的版本是torch1.10.1,CUDA版本是10.1。
但是我在官网下载的是torch1.10.1+cu102,所以报错。
然而,我去翻官网,官网没有torch1.10.1+cu101这个组合可还行…
这只有降pytorch的版本了。
重装pytorch+CUDA
一条命令即可:
pip3 install torch==1.8.1+cu101 torchvision==0.9.1+cu101 -f https://download.py torch.org/whl/cu101/torch_stable.html
重新运行代码未报错。
成功。
本文介绍了解决PyTorch与CUDA版本不匹配导致的运行错误的方法。通过检查并调整PyTorch和CUDA版本,确保两者兼容,最终解决了运行时报错的问题。
8万+

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



