当接触到Linux命令行,我才真正意义上成为一名程序员!跑深度学习图像分类论文时,为了满足环境的需求,通常需要切换Cuda版本,重装Cuda太麻烦!!!
下面介绍一种软连接的方法来实现Cuda版本之间的快速切换(顺便说一下,Nvida驱动要高一点,才能与不同版本的Cuda兼容)!
注:下面的代码我都是以root身份运行的、默认您已经安装了较高版本的驱动!
1、 首先介绍cuda 的下载与安装(以cuda9.0为例)
https://developer.nvidia.com/cuda-toolkit-archive
选择对应型号的cuda:
下载后进入所在目录,运行代码:
sh ./cuda_9.0.176_384.81_linux.run # 安装 cuda_9.0.176_384.81_linux.run
一些安装过程中的选项以及安装结果:
解释:
Do you accept the previously read EULA?
accept/decline/quit: accept
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?
(y)es/(n)o/(q)uit: n # 如果在这之前已经安装好更高版本的显卡驱动就不需要再重复安装,如果需要重复安装就选择 yes,此外还需要关闭图形界面。
Install the CUDA 9.0 Toolkit?
(y)es/(n)o/(q)uit: yes
Enter Toolkit Location
[ default is /usr/local/cuda-9.0 ]: # 一般选择默认即可,也可以选择安装在其他目录,在需要用的时候指向该目录或者使用软连接 link 到 /usr/local/cuda。
Do you want to install a symbolic link at /usr/local/cuda? # 是否将安装目录通过软连接的方式 link 到 /usr/local/cuda,yes or no 都可以,取决于你是否使用 /usr/local/cuda 为默认的 cuda 目录(因为软链方便管理不同版本的cuda,所以推荐使用)。
(y)es/(n)o/(q)uit: yes
Install the CUDA 9.0 Samples?
(y)es/(n)o/(q)uit: no
刚才cuda已经软连接到9.0版本了,现在演示以下如何软链接到其他版本的cuda(cuda-10.1为例):
然后进入Cuda的安装目录(我的Cuda默认安装在这个目录下的):
cd /usr/local/
这种方式看不出软连接:
换一种方式ll
:
可看到cuda文件夹后有一个箭头,表示指向cuda-10.2,就是软链接的意思!
那么如何把cuda软连接到cuda-10.1呢?
首先删除原有的软连接:
sudo rm -rf cuda
建立新的软连接:
sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda
这样cuda就软连接到cuda-10.1了!
测试一下:
Linux命令行的强大!你爱了吗
附NVIDIA驱动、Cuda Toolkit、cudnn下载链接:
NVIDIA驱动:
https://www.nvidia.cn/Download/index.aspx
Cuda Toolkit:
https://developer.nvidia.com/cuda-toolkit-archive3
cudnn:
https://developer.nvidia.com/rdp/cudnn-archive