CentOS7 安装 NVIDIA Container Toolkit

1. 安装containerd.io

   sudo yum install -y https://download.docker.com/linux/centos/7/x86_64/stable/Packages/containerd.io-1.4.3-3.1.el7.x86_64.rpm
   

2. 安装docker-ce

   sudo yum install docker-ce -y
   

3. 使用以下命令确保 Docker 服务正在运行：

   sudo systemctl --now enable docker
   

4. 设置存储库和 GPG 密钥：

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

5. 将experimental分支添加到存储库列表中：

   sudo yum-config-manager --enable libnvidia-container-experimental
   

6. 安装nvidia-docker2

   sudo yum clean expire-cache
   sudo yum install -y nvidia-docker2
   

7. 重启Docker

   sudo systemctl restart docker
   

8. 使用 --gpus all 验证是否可以在容器中运行nvidia-smi

   sudo docker run -it --gpus all <你的容器id> /bin/bash
   

参考