Ubuntu18.04--install NVIDIA-Driver CUDA cuDNN

系统环境：Ubuntu18.04
显卡型号：NVIDIA GeForce RTX 2080 SUPER
架构信息：Turing（CUDA 10 and later），SM75 or SM_75, compute_75 –
Driver Version：470.129.06
CUDA Version：11.4
CUDNN_VERSION：8204 (8.2.4)
Host compiler version : GCC 7.5.0

驱动安装#

两种安装方式：

• Ubuntu更新软安装 - 打开Software & Updates软件 -> Additional Drivers -> 等待搜索完毕 -> 选择NVIDIA Corporation下面的'Using NVIDIA Server Driver metapackage from nvidia-driver-470-server(proprietary)' -> 点击Apply Changes
• NVIDIA官网驱动搜索下载安装，注意驱动版本与显卡的兼容性

CUDA安装#

terminal输入nvidia-smi查看最高支持的CUDA版本号

我的是11.4（即显示信息表格第一行的最右边哪个CUDA Version：11.4，表示此驱动最高支持的CUDA版本）

在NVIDIA-CUDA下载页面查看需要下载的CUDA版本，命令行下载安装

# 下载
wget https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux.run
# 安装，回车之后需耐心等待才会出来安装界面，空格取消勾选驱动安装，其他全选
sudo sh cuda_11.4.4_470.82.01_linux.run

添加环境变量-参考
vim ~/.bashrc 文件末尾添加：

# Add the related bin and lib for cuda toolkit
export PATH="/usr/local/cuda-11.4/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH"

然后source ~/.bashrc
如果使用了zsh则会报错
解决方法：exec bash 切换到bash再source ~/.bashrc
但是此种方法只能在bash上使用，切换回zsh依然找不到，说明zsh的配置文件.zshrc与bash的配置文件.bashrc是相对独立的需要额外在.zshrc里面配置
vim ~/.zshrc 文件末尾添加相同信息，再source ~/.zshrc

检查CUDA是否安装成功#

1 - nvcc --version
2 - 编译测试设备 deviceQuery(路径仅供参考)

cd /usr/local/cuda-11.4/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
# 结果为：Result = PASS代表通过无问题

3 - 编译并测试带宽 bandwidthTest

cd ../bandwidthTest
sudo make
./bandwidthTest
# 结果为：Result = PASS代表通过无问题

cuDNN安装#

cuDNN Archive 官网 查看相关版本，下载安装包安装

显卡驱动和cuda版本存在一一对应的版本范围，nvidia-smi查看驱动版本470兼容的最高cuda版本为11.4，所以下载cuDNN v8.2.4 for CUDA 11.4即可，下载三个文件：
cuDNN Runtime Library for Ubuntu18.04 x86_64 (Deb)
cuDNN Developer Library for Ubuntu18.04 x86_64 (Deb)
cuDNN Code Samples and User Guide for Ubuntu18.04 x86_64 (Deb)
安装命令：
sudo dpkg -i libcudnn8_8.2.4.15-1+cuda11.4_amd64.deb
sudo dpkg -i libcudnn8-dev_8.2.4.15-1+cuda11.4_amd64.deb
sudo dpkg -i libcudnn8-samples_8.2.4.15-1+cuda11.4_amd64.deb

检查cuDNN安装成功#

cp -r /usr/src/cudnn_samples_v8 ~
cd ~/cudnn_samples_v8
cd mnistCUDNN
make clean
make
./mnistCUDNN
# 结果为Test passed！代表cuDNN安装成功

编译mnistCUDNN时出错：fatal error: FreeImage.h: No such file or directory
sudo apt-get install libfreeimage3 libfreeimage-dev

相关参考链接：#

• Matching CUDA arch and CUDA gencode for various NVIDIA architectures
• NVIDIA 驱动搜索页面 https://www.nvidia.com/Download/Find.aspx?lang=en-us
• cuda下载页面 https://developer.nvidia.com/cuda-toolkit-archive
• cuDNN下载页面 https://developer.nvidia.com/rdp/cudnn-archive