在实际工作中,经常会遇到多个项目,要使用不同的框架,而不同的框架可能需要到不同版本的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