ubuntu cuda多版本安装与切换

作者 yatenglg 发布于Posted on

一. cuda多版本说明

cuda可以多个版本共存，并且驱动的cuda版本，可以与运行的cuda版本不同。

通过 nvidia-smi 可以查看驱动的cuda版本。

通过nvcc-v可以查看现在的运行的cuda版本。

二. 多版本cuda安装

下载cuda run文件，管理员运行进行安装。

cuda下载地址，cudnn下载地址，需版本对应。

2.1 cuda安装

取消驱动安装（显卡已经装过驱动了，不要再装驱动）

安装成功后，/usr/local下存在多个版本的cuda

2.2 cudnn安装

cudnn直接安装到/usr/local/cuda-(对应版本) 的文件夹下，便于随时切换。

2.3 环境变量设置

设置cuda环境变量时，不再指定到具体版本的cuda文件夹，便于多版本cuda切换。

# cuda path
# export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
# export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
# export CUDA_HOME=/usr/local/cuda-10.2

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda

3.3 多版本cuda切换

查看当前cuda版本。

查看当前/usr/local/cuda的软链接指向位置。(图中，cuda软链接到cuda-11.5，与当前nvcc查看的cuda版本一致)。

删除当前cuda软链接，并新建软链接指向cuda-10.2。

切换成功。