Ubuntu 20.04 安装NVIDIA显卡驱动+cuda 11.7+cudnn

参考：
https://zhuanlan.zhihu.com/p/59618999
https://blog.csdn.net/linhai1028/article/details/79445722/
Windows+Ubuntu从双系统安装到CUDA cuDNN docker 配置K21

https://blog.csdn.net/qq_45831128/article/details/127060475

https://blog.csdn.net/kunhe0512/article/details/125061911

本机环境

Ubuntu 20.04
英伟达 RTX 3050 Ti 显卡

1 相关查询命令

1）Ubuntu下查看Nvidia显卡的详细信息：

nvidia-smi
1

2）查看cuda 版本：

nvcc -V
1

3）查看cudnn版本：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
1

4）在终端输入命令，实时查看GPU的使用情况：

watch -n 1 nvidia-smi
1

5）在Ubuntu下查询显卡型号

nvidia-smi -a |grep -i Product\ Name
1

---

一、Ubuntu 20.04 安装NVIDIA显卡驱动

下载 NVIDIA-Linux-x86_64-515.57.run
这个版本驱动比较稳定。

二、Ubuntu 20.04 安装cuda 11.7

1、安装CUDA 11.7

cuda 下载地址：https://developer.nvidia.com/zh-cn/cuda-downloads

第二步就是安装CUDA。这里我们安装CUDA最新的11.7版本。进入Nvidia cuda下载地址：
Nvidia cuda11.7 下载地址

我们这里选择的Ubuntu 20.04下的deb安装类型。

然后依次执行下述官方提供的几条命令：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-11-7-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda
1
2
3
4
5
6
7

或者用非root用户安装
参考: 非root用户安装cuda与cudnn

3、配置CUDA环境变量

安装完成之后，加入到环境变量，可用gedit

gedit ~/.bashrc
1

在.bashrc末尾添加两行环境变量

export PATH=$PATH:/usr/local/cuda-11.7/bin
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}  #根据CUDA版本更换路径
1
2

最后，使环境变量生效

source ~/.bashrc
1

测试

nvcc -V
1

~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Tue_May__3_18:49:52_PDT_2022
Cuda compilation tools, release 11.7, V11.7.64
Build cuda_11.7.r11.7/compiler.31294372_0

1
2
3
4
5
6
7

三、cudnn8.4 安装

cudnn下载地址：https://developer.nvidia.com/rdp/cudnn-archive

需要提前注册，找到对应版本，下载cuDNN Library for Linux压缩包即可

解压压缩包，并终端进入文件夹，执行以下

#进入解压目录
cd cudnn-linux-x86_64-8.4.0.27_cuda11.6-archive
# 复制cudnn头文件
sudo cp ./include/* /usr/local/cuda-11.7/include/  
# 复制cudnn的库
sudo cp ./lib/* /usr/local/cuda-11.7/lib64/ 

#给予权限
sudo chmod a+r /usr/local/cuda-11.7/include/cudnn*.h 
sudo chmod a+r /usr/local/cuda-11.7/lib64/libcudnn*
1
2
3
4
5
6
7
8
9
10

（3）测试

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