CentOS 7 上安装 NVIDIA Docker

ManchiBB2024-09-10 13:32

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

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

添加 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

安装 nvidia-docker2 ：安装 nvidia-docker2 包，它提供了 Docker 运行时的 NVIDIA 支持。
复制代码
```
sudo yum install -y nvidia-docker2
```
重启 Docker 服务：安装完成后，您需要重启 Docker 服务以使更改生效。
复制代码
```
sudo systemctl restart docker
```
验证安装：验证 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 的要求，并且您有足够的权限来安装软件和重启服务。