Ubuntu更新显卡驱动及安装多版本cuda

前言:

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显卡去掉(按回车去掉),其他都是默认,按回车就行。

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值