Ubuntu 20.04：一站式 CUDA 10.1 和 cuDNN 安装指南

背景：

NVIDIA 的 CUDA 并行计算平台和 cuDNN 库是现代 AI 和深度学习应用的基石。如果您希望在 Ubuntu 20.04 系统上充分利用这些强大工具，那么本指南将为您提供必要的步骤，让您快速上手。

1. 系统准备

在开始安装之前，请确保您的系统符合以下要求：

• Ubuntu 20.04 操作系统
• NVIDIA 显卡，支持 CUDA 10.1
• 互联网连接

2. 安装 NVIDIA 驱动程序

第一步是安装 NVIDIA 驱动程序，以确保您的显卡能够与 CUDA 10.1 配合使用。

1. 打开终端并输入以下命令以更新您的软件包列表：

sudo apt update

2. 安装附加驱动程序软件包：

sudo apt install nvidia-driver-450

3. 重新启动您的系统以使更改生效。

3. 安装 CUDA 10.1

安装 NVIDIA 驱动程序后，就可以安装 CUDA 10.1 了。

1. 下载 CUDA 10.1 安装程序：

   • 前往 NVIDIA CUDA Toolkit 下载页面：https://developer.nvidia.com/cuda-downloads
   • 选择 Linux 操作系统并下载 runfile（local 安装程序）

2. 将下载的文件移动到您希望安装 CUDA 的目录中。

3. 在终端中导航到该目录并运行以下命令开始安装：

sudo sh cuda_10.1.243_418.87.00_linux.run

4. 按照安装程序提示进行操作，接受许可协议并选择要安装的组件。建议选择“仅安装驱动程序”。

5. 安装完成后，重新启动您的系统。

4. 安装 cuDNN

接下来，我们需要安装 cuDNN，这是一个加速深度学习训练和推理的库。

1. 从 NVIDIA cuDNN 下载页面下载 cuDNN：https://developer.nvidia.com/cudnn

   • 选择 Ubuntu 20.04 和 CUDA 10.1
   • 下载适用于您系统架构的文件（x86_64 或 aarch64）

2. 将下载的文件移动到您的主目录中。

3. 打开终端并导航到您的主目录。

4. 解压 cuDNN 文件：

tar -xzvf cudnn-10.1-linux-x64-v8.0.5.39.tgz

5. 将 cuDNN 库文件移动到适当的位置：

sudo cp cuda/include/cudnn.* /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

5. 验证安装

安装完成后，让我们验证一切是否正常工作。

1. 打开终端并输入以下命令检查 CUDA 版本：

nvidia-smi

您应该看到类似以下内容：

NVIDIA-SMI 450.87.00   Driver Version: 450.87.00   CUDA Version: 10.1

2. 验证 cuDNN 安装：

python3 -c "import tensorflow as tf; print(tf.test.is_gpu_available())"

您应该看到 True，表示 cuDNN 已正确安装。

结论

恭喜！您已成功在 Ubuntu 20.04 系统上安装了 CUDA 10.1 和 cuDNN。现在，您可以充分利用 NVIDIA 的强大并行计算能力，开启您的 AI 和深度学习探索之旅。

请注意，本指南针对的是具有 CUDA 能力的 NVIDIA 显卡。如果您使用的是 AMD 显卡，则需要遵循不同的安装步骤。

如果您在安装过程中遇到任何问题，请随时在评论区提出问题，我会尽我所能提供帮助。