Ubuntu20.04 显卡驱动 CUDA CUDNN 官网安装教程

0. 说明

更多更新，在我的个人主页的该文章。

这是2022年5月4日成功安装nvidia driver&cuda11.6&cudnn的案例，仅供参考。
每项包含安装和验证两个步骤，每个小标题都是对应的指导地址，请点击之。

我写了一个自动化shell脚本，参见

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

安装

1. Navigate to your <cudnnpath> directory containing the cuDNN Debian local installer file.
   下载 deb文件，登录拿qq、wechat就行
2. 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
3. Import the CUDA GPG key.
   sudo apt-key add /var/cudnn-local-repo-*/7fa2af80.pub

4. 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需要的版本号，

5. Install the runtime library.
   sudo apt-get install libcudnn8=8.x.x.x-1+cudaX.Y
6. Install the developer library.
   sudo apt-get install libcudnn8-dev=8.x.x.x-1+cudaX.Y
7. Install the code samples and the cuDNN library documentation.
   sudo apt-get install libcudnn8-samples=8.x.x.x-1+cudaX.Y

验证

1. Copy the cuDNN samples to a writable path.
   cp -r /usr/src/cudnn_samples_v8/ $HOME
2. Go to the writable path.
   cd $HOME/cudnn_samples_v8/mnistCUDNN

3. 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
   

4. Run the mnistCUDNN sample.
   ./mnistCUDNN
   如果没问题的话会显示：
   Test passed!

恭喜！到此已经完成安装！ 你已经拥有nvidia driver & cuda11.6 & cudnn8.4.0.27-1！ XD

其他

Google pinyin
搜狗是shit

如果有帮你节约时间，不妨帮我点个赞哈，感谢支持XD