CentOS 7 上安装 NVIDIA Docker
在 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 的要求,并且您有足够的权限来安装软件和重启服务。