前言

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

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐