问题:遇到如下报错,
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
重新运行代码未报错。
成功。