ubuntu下CUDA以及cudnn安装

下载库

确定自己所需要的cuda版本以及对应的cudnn版本。
CUDA下载链接：
https://developer.nvidia.com/cuda-toolkit-archive
cudnn下载链接：
https://developer.nvidia.cn/rdp/cudnn-download
*其中cudnn需要注册开发者账号进行下载

安装CUDA

本文以安装cuda11.7为例：

打开对应网站，选择自己所需版本，按照窗口内的指令输入即可，建议安装deb（local）

最后在自己的bash内添加cuda对应的path

gedit ~/.bashrc

添加如下命令

export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

输入以下命令更新bash

source ~/.bashrc

检测cuda安装

nvcc -V

效果显著！！！

安装cudnn

此处假设已经下载好了cudnn的deb文件

根据官网安装文件指南：https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html

该部分来完成安装，笔者将按照本机上的cudnn库演示安装

sudo dpkg -i cudnn-local-repo-ubuntu2004-8.5.0.96_1.0-1_amd64.deb
#后者需要修改为自己下载的cudnn库文件

复制该指令运行

sudo cp /var/cudnn-local-repo-ubuntu2004-8.5.0.96/cudnn-local-0579404E-keyring.gpg /usr/share/keyrings/

更新源，也就是把这个deb的内容添加到apt里

sudo apt-get update

此时我们能在apt内找到相应的库了

此时不用按照官方教程内的语句来逐字逐句输入了，直接安装这几个就行了；

sudo apt-get install libcudnn8 libcudnn8-dev libcudnn8-samples

优雅完成！！！！