在使用 yum 安装 NVIDIA Docker 时，可以通过指定 nvidia-docker.repo 文件来确保从正确的存储库安装。以下是详细步骤：
- 下载并安装 NVIDIA Docker 的 YUM 存储库文件
首先，需要下载 NVIDIA 官方提供的 nvidia-docker.repo 文件，并将其放置到 /etc/yum.repos.d/ 目录下。
1.1 下载 nvidia-docker.repo 文件
运行以下命令，从 NVIDIA 官方网站下载存储库文件：
bash
复制
distribution=$(. /etc/os-release; echo I D ID IDVERSION_ID) # 自动获取系统版本
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo
如果需要手动指定系统版本，可以替换 d i s t r i b u t i o n 的值。例如： b a s h 复制 c u r l − s − L h t t p s : / / n v i d i a . g i t h u b . i o / n v i d i a − d o c k e r / c e n t o s 7 / n v i d i a − d o c k e r . r e p o ∣ s u d o t e e / e t c / y u m . r e p o s . d / n v i d i a − d o c k e r . r e p o 1.2 验证存储库文件检查 / e t c / y u m . r e p o s . d / n v i d i a − d o c k e r . r e p o 文件是否已正确创建： b a s h 复制 c a t / e t c / y u m . r e p o s . d / n v i d i a − d o c k e r . r e p o 您应该能看到类似以下内容： i n i 复制 [ n v i d i a − d o c k e r ] n a m e = N V I D I A D o c k e r b a s e u r l = h t t p s : / / n v i d i a . g i t h u b . i o / n v i d i a − d o c k e r / c e n t o s 7 / distribution 的值。例如： bash 复制 curl -s -L https://nvidia.github.io/nvidia-docker/centos7/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo 1.2 验证存储库文件 检查 /etc/yum.repos.d/nvidia-docker.repo 文件是否已正确创建： bash 复制 cat /etc/yum.repos.d/nvidia-docker.repo 您应该能看到类似以下内容： ini 复制 [nvidia-docker] name=NVIDIA Docker baseurl=https://nvidia.github.io/nvidia-docker/centos7/ distribution的值。例如：bash复制curl−s−Lhttps://nvidia.github.io/nvidia−docker/centos7/nvidia−docker.repo∣sudotee/etc/yum.repos.d/nvidia−docker.repo1.2验证存储库文件检查/etc/yum.repos.d/nvidia−docker.repo文件是否已正确创建：bash复制cat/etc/yum.repos.d/nvidia−docker.repo您应该能看到类似以下内容：ini复制[nvidia−docker]name=NVIDIADockerbaseurl=https://nvidia.github.io/nvidia−docker/centos7/basearch/
enabled=1
gpgcheck=1
gpgkey=https://nvidia.github.io/nvidia-docker/gpgkey - 安装 NVIDIA Docker
在添加存储库文件后，使用 yum 安装 NVIDIA Docker：
2.1 安装 nvidia-docker2 包
运行以下命令安装 NVIDIA Docker：
bash
复制
sudo yum install -y nvidia-docker2
2.2 配置 Docker 使用 NVIDIA 运行时
安装完成后，需要配置 Docker 使用 NVIDIA 运行时。运行以下命令：
bash
复制
sudo nvidia-ctk runtime configure --runtime=docker
然后重启 Docker 服务：
bash
复制
sudo systemctl restart docker - 验证安装
运行以下命令验证 NVIDIA Docker 是否安装成功：
bash
复制
docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi
如果安装成功，您将看到 GPU 的状态信息。
注意事项
确保系统已安装 NVIDIA 驱动
在安装 NVIDIA Docker 之前，请确保您的系统已安装最新版本的 NVIDIA 驱动，并且驱动版本与 CUDA 版本兼容。可以使用以下命令检查驱动版本：
bash
复制
nvidia-smi
系统版本匹配
确保下载的 nvidia-docker.repo 文件与您的 CentOS 版本匹配。例如，如果您使用的是 CentOS 7，则需要下载 centos7 的存储库文件。
依赖关系
如果在安装过程中遇到依赖问题，请确保您的系统已更新到最新版本：
bash
复制
sudo yum update -y
通过以上步骤，您可以使用指定的 nvidia-docker.repo 文件安装并配置 NVIDIA Docker。
以上内容为ai自动生成，可以使用，注意服务器会有的多条repow文件，需要将多余的文件挪走即可
