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

2022-01-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 下即可。

上一篇下一篇

猜你喜欢

热点阅读