CentOS 7 上安装 NVIDIA Docker

在 CentOS 7 上安装 NVIDIA Docker 需要几个步骤。以下是安装 NVIDIA Docker 的基本步骤：

1. 安装 NVIDIA 驱动： 确保您的系统已经安装了 NVIDIA 驱动。如果没有，请先安装驱动。

2. 添加 NVIDIA Docker 仓库： 添加 NVIDIA Docker 的官方仓库，以便安装最新版本的 NVIDIA Docker。

   distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
   curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo

3. 安装 nvidia-docker2： 安装 nvidia-docker2 包，它提供了 Docker 运行时的 NVIDIA 支持。

   sudo yum install -y nvidia-docker2

4. 重启 Docker 服务： 安装完成后，您需要重启 Docker 服务以使更改生效。

   sudo systemctl restart docker

5. 验证安装： 验证 NVIDIA Docker 是否正确安装，可以通过运行一个带有 GPU 支持的基础容器来测试。

   sudo docker run --rm --gpus all nvidia/cuda:10.0-base nvidia-smi

   如果一切正常，上述命令将输出 NVIDIA GPU 的相关信息。

请注意，上述命令中的 CUDA 版本（例如 10.0-base）可能需要根据您的需求进行更改。您可以从 NVIDIA 官方 Docker Hub 仓库中找到支持的 CUDA 版本。

在进行这些步骤之前，请确保您的系统满足 NVIDIA Docker 的要求，并且您有足够的权限来安装软件和重启服务。