centos nvidia-docker2安装及使用

nv

idia-docker2安装

1.设置所需仓科库：

curl -s -L https://nvidia.github.io/libnvidia-container/stable/rpm/nvidia-container-toolkit.repo | \
  sudo tee /etc/yum.repos.d/nvidia-container-toolkit.repo

可选：您还可以选择配置存储库，以便使用实验性软件包

sudo yum-config-manager --enable nvidia-container-toolkit-experimental

2.安装NVIDIA容器工具包

sudo yum install -y nvidia-container-toolkit

配置：

修改镜像源：

sudo vim /etc/docker/daemon.json

{
    "runtimes": {
        "nvidia": {
            "path": "nvidia-container-runtime",
            "runtimeArgs": []
        }
    },
    "registry-mirrors": [
        "https://registry.docker-cn.com"
    ]
}