nvidia-container-toolkit

安装

1
# 先配置存储库
2
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
3
  && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
4
    sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
5
    sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
6

7
# 更新软件包
8
sudo apt-get update
9

10
# 安装
11
export NVIDIA_CONTAINER_TOOLKIT_VERSION=1.17.8-1
12
  sudo apt-get install -y \
13
      nvidia-container-toolkit=${NVIDIA_CONTAINER_TOOLKIT_VERSION} \
14
      nvidia-container-toolkit-base=${NVIDIA_CONTAINER_TOOLKIT_VERSION} \
15
      libnvidia-container-tools=${NVIDIA_CONTAINER_TOOLKIT_VERSION} \
16
      libnvidia-container1=${NVIDIA_CONTAINER_TOOLKIT_VERSION}

因为网络的原因第一步骤可能会下载失败，下面是 nvidia-container-toolkit.list 的内容，保存到 /etc/apt/sources.list.d 目录下:

1
deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://nvidia.github.io/libnvidia-container/stable/deb/$(ARCH) /
2
#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://nvidia.github.io/libnvidia-container/experimental/deb/$(ARCH) /

使用

大部分安装 nvidia-container-toolkit 驱动都是为了让 [[Docker]] 能支持 GPU，可以通过 nvida-ctk 进行配置:

1
sudo nvidia-ctk runtime configure --runtime=docker

该命令会自动在 /etc/docker/daemon.json 加上:

1
{
2
    "runtimes": {
3
        "nvidia": {
4
            "args": [],
5
            "path": "nvidia-container-runtime"
6
        }
7
    }

然后通过 [[systemctl]] 命令重启 Docker 进程:

1
sudo systemctl restart docker

参考

Installing the NVIDIA Container Toolkit — NVIDIA Container Toolkit