前言:
win下安装高版本NVIDIA驱动,在conda环境下可以通过安装cudatoolkit自由切换不同版本的cuda。
但是在Ubuntu上实践时,由于最开始NVIDIA驱动过低,导致conda环境下切换失败,因此本文将首先对NVIDIA驱动进行升级。
安装了NVIDIA驱动,通过nvidia-smi可以查看该驱动最高支持的cuda版本
1. 下载显卡驱动
网址:https://www.nvidia.cn/Download/index.aspx?lang=cn
2. 安装显卡驱动
参考https://blog.csdn.net/feynman233/article/details/79187490中的ppa安装法:
ctrl+alt+F1进入文本模式:
关闭图形界面:
sudo /etc/init.d/lightdm stop
卸载旧的驱动:
sudo apt-get --purge remove nvidia-*
安装新驱动:
sudo sh NVIDIA-Linux-x86_64-xxx.run
最后重启OK
参考:https://www.cnblogs.com/chua-n/p/13208398.html
3. 安装cuda
参考https://blog.csdn.net/feynman233/article/details/79187490和https://blog.csdn.net/yinxingtianxia/article/details/80462892
cuda下载地址:https://developer.nvidia.com/cuda-downloads
选取runfile文件安装方式
安装过程中在建立软链接时需要注意一下,如果你是第一次安装cuda,那么毫无疑问输入y(yes),但是如果你是安装额外版本的cuda,是否选择y(yes)就要看你的具体需求而定,简言之,就是如果你希望启用当前安装的cuda版本,就选y,如果你只是想安装这个版本,而暂时还不想启用该版本时,就选n。
4. cuda多个版本的切换
在安装了多个cuda版本后,可以在/usr/local/目录下查看自己安装的cuda版本,cuda是一个软链接,它指向我们指定的cuda版本(注意上面在设置环境变量时,使用的是cuda,而不是cuda-9.0和cuda-9.1,这主要是为了方便我们切换cuda版本,可以让我们不用每次都去该环境变量的值)
可以使用stat命令查看当前cuda软链接指向的哪个cuda版本。
当我们想使用别的cuda版本时,只需要删除该软链接,然后重新建立指向cuda新版本的软链接即可(注意名称还是cuda,因为要与bashrc文件里设置的保持一致)
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-9.1 /usr/local/cuda
不过,有时,改完软连接,似乎cmd下nvcc -V
还是旧版本cuda,综合考虑,似乎还是直接改~/.bashrc文件
更直接简便。(也有可能与~/.bashrc文件
缺少这句话有关:export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
)
参考:https://blog.csdn.net/ksws0292756/article/details/80120561
source ~/.bashrc
之后需要打开新终端激活
anaconda 配置不同版本cuda问题(似乎不太可):https://blog.csdn.net/LMM_AI/article/details/106857663
安装cuda11时,在很多X选择的时候,把第一个Nvidia显卡去掉(按回车去掉),其他都是默认,按回车就行。