在实际工作中,经常会遇到多个项目,要使用不同的框架,而不同的框架可能需要到不同版本的cuda,这个时候就需要安装多个cuda,并且随时可以进行切换版本。本文主要是记录如何对cuda版本进行切换。

设置系统使用的cuda路径

  将目录 ~/.bashrc 或 ~/.zshrc 下与cuda相关的路径都改为软连接路径 /usr/local/cuda/ 而不使用 /usr/local/cuda-8.0/ 或 /usr/local/cuda-9.0/

创建指向cuda版本的软连接

在使用cuda的过程中,系统会使用你软连接到的cuda版本。

因此,我们只需要在每次使用的时候,将软连接指定到需要使用的版本文件夹。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
#在切换cuda版本时
rm -rf /usr/local/cuda #删除之前创建的软链接
sudo ln -s /usr/local/cuda-8.0/ /usr/local/cuda
nvcc --version #查看当前 cuda 版本

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Mon_Jan_23_12:24:11_CST_2017
Cuda compilation tools, release 8.0, V8.0.62

#cuda8.0 切换到 cuda9.0
rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-9.0/ /usr/local/cuda
nvcc --version