linux下安装多版本cuda和cudnn

    科技2025-01-02  24

    前言

    随意切换cuda-8.0和cuda-10.1。

    查看cuda版本和cudnn版本

    查看cuda版本 nvcc –V 或者 cat /usr/local/cuda/version.txt 查看cudnn版本 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

    那我们来看一下我现在的版本:cuda-10.1和cudnn-7.65

    切换版本

    首先需要我们安装好cuda和cudnn: cuda官网:https://developer.nvidia.com/cuda-toolkit-archive cudnn官网:https://developer.nvidia.com/cudnn 这里就不做介绍了哈,网上很多教程

    cuda版本的自由切换

    我们要做的就是cuda版本的自由切换。命令行sudo gedit ~/.bashrc打开.bashrc,也许你已经装过cuda-10.1,则.bashrc下应该有如下行:

    export PATH="$PATH:/usr/local/cuda-10.1/bin" export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64/" export LIBRARY_PATH="$LIBRARY_PATH:/usr/local/cuda-10.1/lib64"

    我们将export中所有的cuda-10.1更改为cuda(因为需要建立一个软链接,所以写成cuda比较方便和下面的cuda文件夹对应):

    export PATH="$PATH:/usr/local/cuda/bin" export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64/" export LIBRARY_PATH="$LIBRARY_PATH:/usr/local/cuda/lib64"

    那我们来看一下我现在的版本:cuda-10.1和cudnn-7.65 然后通过创建软连接,实现cuda版本的自由切换:

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

    现在版本变为:cuda-8.0和cudnn-7.03版本 /usr/local/文件夹下的文件:

    Processed: 0.012, SQL: 8