Ubuntu22.04安装CUDA12.1 cuDNN8.9.7 pytorch2.2.2

当前安装的电脑配置:

电脑名称：暗影精灵5

系统名称：Ubuntu22.04.4 LTS

操作系统类型：64位

内存：8.0Gib

处理器：Intel®Core™i7-9750H CPU @ 2.60GHz x 12

显卡：NVIDIA Corporation TU117M [GeForce GTX 1650 Mobile / Max-Q] Mesa I...

安装前准备：需要安装对应的NVIDIA驱动程序、配置CUDA依赖环境g++ gcc make

sudo apt-get install vim g++ gcc make python3-pip

需要检查是否安装了NVIDIA驱动程序，如果未安装下载安装。

Download the latest official NVIDIA drivers：NVIDIA驱动程序

1. 安装CUDA

1.1 打开下面官方链接下载CUDA

CUDA Toolkit Archive：CUDA工具包存档

1.1.1 选择CUDA Toolkit 12.1.0（可以根据自己的需要选择对应的版本）

1.1.2 根据自己的配置选择对应的系统、架构、版本和安装方式：

1.1.3 完成上述选择后会出现安装命令（你可以根据自己的情况选择安装方式如deb(local)或者deb(network)方式，我选择runfile(local)方式命令较为少只有两行）

1.1.3.1 将命令在终端中运行

wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run
sudo sh cuda_12.1.0_530.30.02_linux.run

wget下载完后在运行上述sudo sh cuda_12.1.0_530.30.02_linux.run后会显示选择页面按照如下操作:

①选择使用方向键（Up/Down)上下选择 Continue选项回车。

②（接受用户协议）输入accept，按下回车键

③同意后，跳转到下面页面，因为在装系统的时候就已经安装过显卡驱动了在Driver回车取消勾选

④完成操作后选中Install回车开始下载

⑤下载完成后输入下面命令配置环境变量

vim ~/.bashrc

⑥在文件末尾增加两个路径下面第一个是官方文档提供的代码，第二个是网络上普遍使用的代码测试都可用

export PATH=/usr/local/cuda-12.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.0/lib64\
                         ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

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

⑦输入命令重新更改环境变量

source ~/.bashrc

⑧测试cuda是否安装成功

nvcc -V

2.安装cuDNN

2.1打开官方连接下载cuDNN

https://developer.nvidia.com/rdp/cudnn-archive#a-collapse805-111:cuDNN下载

2.2根据自己安装的cuda版本和自己的需要选择cuDNN版本，我在这选择cuDNN8.9.4 for CUDA 12.x

2.3根据自己电脑的版本系统以及自己喜欢的安装方式选择下载，我在这选择deb包。在这需要登录你的NVDIA账号，如果没有可以自己注册一个。

2.4在这提供cuDNN文档存档地址：NVIDIA cuDNN 文档在这我使用的8.9.4，如果使用的9.0.0以后文档请点击后面的链接（文档连接（可选择最新还是以前的))

2.5找到自己选择的安装方式，我在这使用的deb方式

下载 Debian 本地存储库安装包。在发布 按照以下命令，必须将 X.Y 和 8.x.x.x 替换为特定的 CUDA 和 cuDNN 版本。

①导航到包含 cuDNN 的下载目录 Debian 本地安装程序文件。

②启用本地存储库。或

sudo dpkg -i cudnn-local-repo-$distro-8.x.x.x_1.0-1_amd64.deb

sudo dpkg -i cudnn-local-repo-$distro-8.x.x.x_1.0-1_arm64.deb

其中 是 、 、 或 。$distroubuntu1804ubuntu2004ubuntu2204debian11

③导入 CUDA GPG 密钥。

sudo cp /var/cudnn-local-repo-*/cudnn-local-*-keyring.gpg /usr/share/keyrings/

④刷新存储库元数据。

sudo apt-get update

⑤安装运行时库。

这里的libcudnn8和cuda版本的配对是指定的，可通过apt-cache policy libcudnn8命令查看

sudo apt-get install libcudnn8=8.x.x.x-1+cudaX.Y

⑥安装开发人员库。

sudo apt-get install libcudnn8-dev=8.x.x.x-1+cudaX.Y

⑦安装代码示例。

sudo apt-get install libcudnn8-samples=8.x.x.x-1+cudaX.Y

2.6测试安装的cuDNN是否成功

要验证 cuDNN 是否已安装并正常运行，请编译位于 Debian 文件的 /usr/src/cudnn_samples_v8 目录中的 mnistCUDNN 示例。

①将 cuDNN 示例复制到可写路径。

cp -r /usr/src/cudnn_samples_v8/ $HOME

②转到可写路径。

cd  $HOME/cudnn_samples_v8/mnistCUDNN

③编译 mnistCUDNN 示例。

make clean && make

④运行 mnistCUDNN 示例。

./mnistCUDNN

如果 cuDNN 已正确安装并在 Linux 系统上运行，您将看到 类似于 以后：

Test passed!