本教程介绍如何为 Docker 安装 NVIDIA Container Toolkit，使容器支持 GPU 加速，并基于中国科技大学（USTC）镜像源加速安装过程。

1. 添加 NVIDIA 官方稳定版仓库与 GPG 密钥（原始源）

获取系统版本信息：

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)

添加 NVIDIA 的 GPG key：

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -

添加 NVIDIA Docker 源：

curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list \
| sudo tee /etc/apt/sources.list.d/nvidia-docker.list

2. 下载 USTC（中国科技大学）提供的 GPG Key

curl -fsSL https://mirrors.ustc.edu.cn/libnvidia-container/gpgkey \
| sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg

3. 配置 USTC 镜像加速源

将官方源替换为 USTC 镜像：

curl -s -L https://mirrors.ustc.edu.cn/libnvidia-container/stable/deb/nvidia-container-toolkit.list \
| sed 's#deb https://nvidia.github.io#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://mirrors.ustc.edu.cn#g' \
| sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

4. 更新 APT 包列表

sudo apt-get update

5. 安装 NVIDIA Container Toolkit

sudo apt-get install -y nvidia-container-toolkit

6. 验证安装是否成功

nvidia-container-cli --version

7. 配置 NVIDIA 运行时（nvidia-ctk）

sudo nvidia-ctk runtime configure --runtime=docker

8. 重启 Docker 服务

sudo systemctl restart docker