Ubuntu 下不同版本 cuda 和 cudnn 切换

今天药忘吃喽~ 2023-02-11 10:23 3阅读 0赞

# Ubuntu 不同版本 cuda 和 cudnn 切换 #

可能不同的软件对与 cuda 和 cudnn 的版本有不同的要求，所以需要根据具体的要求来切换其版本。

ubuntu 是支持安装多个版本的 cuda 的，需要使用某个的版本时切换就可以了。对于 cudnn 则是直接重装，所谓的重装其实就是 copy 后创建个软链接就可以了。

安装 cuda 比较容易，直接运行下载的 sh 文件就可以了。

sudo sh cuda_9.0.176_384.81_linux.run

## 1. 切换 cuda ##

首先应该确保环境变量的设置不是 hard code，而是指向软链接，这样就可以直接通过修改软链接来修改 cuda 的版本。当然 hard code 的方式可以通过直接修改环境变量来修改 cuda 的版本，也是可以的。

通过如下命令进入环境变量文件

sudo gedit ~/.bashrc

检查关于 cuda 的环境变量设置，如下的设置是 hard code 的，直接指定了 cuda 的版本。

# for cuda9.0
    export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

而下面的设置是指向 /usr/local/cuda 的软链接。

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

对于上面这种情况，通过如下指令直接删除之前的软链接，创建一个指向新的 cuda 版本的软链接即可。

rm -rf /usr/local/cuda  #删除之前创建的软链接
    sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda

修改完之后可以查看 cuda 的版本是否已经修改成功。

nvcc -V

## 2. 切换 cudnn ##

准确的说切换 cudnn 的过程就是重新安装一遍 cudnn，因为 cudnn 的安装本身就是 copy 一些文件到 cuda 中，然后创建软链接。

# 解压cudnn文件
    cd ~/Downloads
    tar zxvf cudnn-9.0-linux-x64-v7.tgz
    
    # 拷贝头文件
    cd cuda/include/
    sudo cp cudnn.h /usr/local/cuda/include/
    
    # 拷贝库文件
    cd cuda/lib64/
    sudo cp lib* /usr/local/cuda/lib64/
    
    # 重新建立软链接
    cd /usr/local/cuda/lib64/
    sudo rm -rf libcudnn.so libcudnn.so.7
    sudo ln -s libcudnn.so.7.0.5 libcudnn.so.7  #注意版本对应
    sudo ln -s libcudnn.so.7 libcudnn.so