ubuntu20.04|ubuntu20.04 显卡驱动 CUDA CUDNN 官网安装教程 ubuntu20.04显卡驱动CUDACUDNN官网安

0. 说明 【ubuntu20.04|ubuntu20.04 显卡驱动 CUDA CUDNN 官网安装教程】这是2022年5月4日成功安装nvidia driver&cuda11.6&cudnn的案例，仅供参考。
每项包含安装和验证两个步骤，每个小标题都是对应的指导地址，请点击之。
1. NVIDIA Driver 安装
推荐510版本（对应cuda11.6）
验证
能在命令行中使用nvidia-smi指令就成功了，如果不行可能是没有重启的原因
2. CUDA(CUDA Toolkit) 安装
主要是看nvidia-smi中的cuda版本，在官网列表中找到自己的版本，按机器的情况选择，就会得到完整的指令（runfile是图形界面，deb是命令行，个人推荐deb）
参考官方教程是deb方式的
验证
进行安装成功验证如下：

$ git clone --depth=1 https://github.com/NVIDIA/cuda-samples.git $ cd cuda-samples/Samples/1_Utilities/deviceQuery $ sudo make $ ./deviceQuery # 最后输出PASS就成功了 ./deviceQuery Starting... CUDA Device Query (Runtime API) version (CUDART static linking) .... .... Result = PASS

似乎需要给.bashrc加内容，但我不清楚作用是什么。

$ vim ~/.bashrc# 在最后加入以下两句，cuda版本号我是11.6，可能需要修改 export PATH=/usr/local/cuda-11.6/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}# 最后 $ source ~/.bashrc

3. CUDNN 安装

Navigate to your directory containing the cuDNN Debian local installer file.
下载 deb文件，登录拿qq、wechat就行
Enable the local repository.
sudo dpkg -i cudnn-local-repo-${OS}-8.x.x.x_1.0-1_amd64.deb
or
sudo dpkg -i cudnn-local-repo-${OS}-8.x.x.x_1.0-1_arm64.deb
Import the CUDA GPG key.
sudo apt-key add /var/cudnn-local-repo-*/7fa2af80.pub
Refresh the repository metadata.
sudo apt-get update

在这一步中，你会看到类似于Get:8 file:/var/cudnn-local-repo-ubuntu2004-8.4.0.27 Release [564 B]
进入这个文件夹，里面是你能选的cudnn版本，也就是步骤5 6 7需要的版本号，
Install the runtime library.
sudo apt-get install libcudnn8=8.x.x.x-1+cudaX.Y
Install the developer library.
sudo apt-get install libcudnn8-dev=8.x.x.x-1+cudaX.Y
Install the code samples and the cuDNN library documentation.
sudo apt-get install libcudnn8-samples=8.x.x.x-1+cudaX.Y

验证

Copy the cuDNN samples to a writable path.
cp -r /usr/src/cudnn_samples_v8/ $HOME
Go to the writable path.
cd $HOME/cudnn_samples_v8/mnistCUDNN
Compile the mnistCUDNN sample.
make clean && make
在这一步我显示了

test.c:1:10: fatal error: FreeImage.h: No such file or directory 1 | #include "FreeImage.h" |^~~~~~~~~~~~~ compilation terminated. >>> WARNING - FreeImage is not set up correctly. Please ensure FreeImage is set up correctly. <<<

需要安装好FreeImage，参考

$ sudo apt-get install g++ freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libglu1-mesa libglu1-mesa-dev $ sudo apt-get install libfreeimage3 libfreeimage-dev
Run the mnistCUDNN sample.
./mnistCUDNN
如果没问题的话会显示：
Test passed!