返回

Ubuntu Server 22.04 LTS 安装 cuDNN：机器学习环境搭建

后端

对于机器学习 (ML) 爱好者和专业人士来说，NVIDIA 的 cuDNN 是一个必不可少的工具。cuDNN 是一个深度学习 GPU 加速库，可显着提高基于 GPU 的深度学习应用程序的性能。在 Ubuntu Server 22.04 LTS 上安装 cuDNN 是构建全面机器学习环境的关键步骤。本教程将逐步指导您完成此过程。

先决条件

  • 运行 Ubuntu Server 22.04 LTS 的服务器
  • NVIDIA GPU
  • CUDA 12.x 已安装

安装依赖项

在开始安装 cuDNN 之前，我们需要安装一些依赖项：

sudo apt update
sudo apt install libcudnn8-dev libcudnn8 libnccl2 libnccl-dev

下载 cuDNN

访问 NVIDIA cuDNN 下载页面并下载适用于 CUDA 12.x 的 cuDNN v8.9.7。确保选择适用于您操作系统（Ubuntu 22.04）的版本。

解压 cuDNN

下载完成后，解压缩 cuDNN 文件：

tar -xzvf cudnn-12.x-linux-x64-v8.9.7.tgz

安装 cuDNN

现在，我们可以将 cuDNN 库文件安装到系统中：

sudo cp cuda/include/cudnn.h /usr/local/include/
sudo cp cuda/lib64/libcudnn* /usr/local/lib/
sudo chmod a+r /usr/local/include/cudnn.h /usr/local/lib/libcudnn*

验证安装

要验证 cuDNN 是否已成功安装，请运行以下命令：

nvcc -V | grep cuDNN

您应该看到以下输出：

...
cuDNN version 8.9.7 ...

配置环境变量

最后，我们需要配置环境变量以使应用程序可以访问 cuDNN：

echo 'export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH' >> ~/.bashrc
echo 'export CUDA_HOME=/usr/local/cuda' >> ~/.bashrc
source ~/.bashrc

结论

通过按照本教程中的步骤，您已成功在 Ubuntu Server 22.04 LTS 上安装了 cuDNN。现在，您可以利用 cuDNN 的强大功能来加速您的机器学习应用程序。