ubuntu 20.04下多版本cuda&cudnn下载与安装

---

前言

Linux Ubuntu 20.04
已经存在cuda 11.3，能够正常使用

---

一、环境检查

1、通过下述命令，检查当前nvidia驱动最高支持的cuda版本：

nvidia-smi

由输出可知，当前驱动为 525.147.05，最高支持的cuda版本为12.0

2、使用stat命令查看当前cuda软链接指向的哪个cuda版本

cd /usr/local/
ls
stat cuda

根据输出可知，当前软链接指向cuda 11.3

---

二、多版本cuda&cudnn下载与安装

cuda下载与安装

1、根据官方链接https://developer.nvidia.com/cuda-toolkit-archive，找到想要下载的cuda版本，以cuda 11.6为例。如下图所示，选择runfile(local)的方式：

在命令行中执行下述指令进行下载

wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux.run

2、安装：下载结束后，执行

sudo sh cuda_11.6.0_510.39.01_linux.run

选择continue

输入accept

接下来会询问安装哪些内容，注意，在这里不要安装驱动！！！

因为当前已经有cuda了，如果只想安装暂时不用的话，就选择no

cudnn下载与安装

1、根据官方链接https://developer.nvidia.com/rdp/cudnn-archive，选择与cuda版本相匹配的cudnn下载tar文件。在这里，我下载的是v8.9.7，可与cuda 11.x等版本适配。

2、解压cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz后，进入该目录，将lib内的文件都复制到目录/usr/local/cuda-11.6/lib64/中，将include内的文件都复制到目录/usr/local/cuda-11.6/include/中

cd ./cudnn-linux-x86_64-8.9.7.29_cuda11-archive
sudo cp ./lib/* /usr/local/cuda-11.6/lib64/
sudo cp ./include/* /usr/local/cuda-11.6/include/

此时，可通过下述指令，判断cudnn是否安装成功

cat /usr/local/cuda-11.6/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

输出以下内容

tianyu@tianyu:~$ cat /usr/local/cuda-11.6/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 9
#define CUDNN_PATCHLEVEL 7
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

/* cannot use constexpr here since this is a C-only file */

多版本cuda切换

通过软链接切换已安装的cuda版本

cd /usr/local/
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-11.6 /usr/local/cuda

由stat cuda指令可查看当前使用的cuda，由输出内容可知，当前正在使用cuda 11.6

tianyu@tianyu:/usr/local$ stat cuda
  File: cuda -> /usr/local/cuda-11.6
  Size: 20        	Blocks: 0          IO Block: 4096   symbolic link
Device: 10302h/66306d	Inode: 30940190    Links: 1
Access: (0777/lrwxrwxrwx)  Uid: (    0/    root)   Gid: (    0/    root)
Access: 2024-03-14 15:34:54.347787279 +0800
Modify: 2024-03-14 15:34:51.447819842 +0800
Change: 2024-03-14 15:34:51.447819842 +0800
 Birth: -

或者使用nvcc -V指令，也可查看当前使用的cuda版本

nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Fri_Dec_17_18:16:03_PST_2021
Cuda compilation tools, release 11.6, V11.6.55
Build cuda_11.6.r11.6/compiler.30794723_0

---

参考

https://blog.csdn.net/yinxingtianxia/article/details/80462892
https://blog.csdn.net/weixin_37926734/article/details/123033286