美文网首页
Ubuntu多用户实现不同版本的CUDA和cuDNN安装

Ubuntu多用户实现不同版本的CUDA和cuDNN安装

作者: 西北小生_ | 来源:发表于2022-01-04 22:04 被阅读0次

    应用场景:Ubuntu多用户原来共享相同的CUDA和cuDNN,现在当前用户下有其它版本的CUDA和cuDNN使用需求。假设原来装的是cuda-10.2并共享所有用户,现在某一个用户要用到cuda-11.3。

    1.安装cuda

    首先下载cuda-11.3的对应版本:

    wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run
    

    然后进行安装:

    sudo sh cuda_11.3.1_465.19.01_linux.run
    

    待安装界面出现后:


    安装界面

    因为已经有显卡驱动了,所以这里一定要把安装驱动这一项取消!

    选中Install选项后会跳转到这一界面:


    image.png

    提示系统中已经存在一个cuda版本在 /usr/local/cuda 目录下,是否更新。这里选择 【No】,然后等几分钟就安装好了。

    待安装cuda-11.3成功后,执行命令:

    sudo vim ~/.bashrc
    

    打开当前用户环境变量界面,并在末尾换行,添加以下语句:

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

    保存并退出。【注意】将其中的11.3改为自己对应的版本!

    执行命令:

    source ~/.bashrc
    

    激活环境变量。至此,cuda就安装完成了。接下来继续安装cudnn。

    2.安装cuDNN

    下载匹配cuda-11.3的cuDNN版本,执行解压命令:

    tar -zvxf cudnn-11.3-linux-x64-v8.2.1.32.tgz
    

    解压cuDNN。【注意】:对应自己的cudnn版本!

    解压后会在当前目录下出现一个名为cuda的文件夹,即为解压后的cuDNN.

    执行命令:

    sudo cp cuda/include/cudnn.h /usr/local/cuda-11.3/include
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.3/lib64
    

    将cudnn的头文件等复制到对应目录下,安装完毕。

    这里的不同之处在于,安装第一个cuda版本时,通常会在 /usr/local 目录下创建一个名为 cuda 的软链接,并将 cudnn 的相应文件复制到该软链接 cuda 目录下。

    而第二次安装 cuda 时并没有创建软链接,而是安装在 /usr/local/cuda-11.3 路径下,并将 cudnn 文件直接复制到相应的安装目录 /usr/local/cuda-11.3 下即可。

    相关文章

      网友评论

          本文标题:Ubuntu多用户实现不同版本的CUDA和cuDNN安装

          本文链接:https://www.haomeiwen.com/subject/cnkccrtx.html