问题描述:
今天想跑一下involution的代码,从GitHub上把代码下下来后,在配置虚拟环境时,安装pytorch时遇到了如下报错:
“The NVIDIA driver on your system is too old”
解决办法:
不需要更新驱动,只需要从pytorch官网重新安装对应的pytorch版本,安装之前用下面命令(linux系统命令)查看自己的cuda版本:
ls -l /usr/local | grep cuda
这里我的CUDA版本是10.1
在官网选择对应的安装方法进行安装即可(conda or pip),我使用的是pip方法下载安装的。
安装默认是官网海外服务器,添加国内镜像源就可以实现快速的download,依次执行:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda install pytorch==1.6.0+cu101 torchvision==0.7.0+cu101