引言
Nvidia Docker该项目已被NVIDIA Container Toolkit取代。此存储库提供的工具已被弃用,并且该存储库已存档。
nvidia-docker
不再支持包装器,并且 NVIDIA Container Toolkit 已进行扩展,允许用户配置 Docker 以使用 NVIDIA Container Runtime 。
NVIDIA 容器工具包允许用户构建和运行 GPU 加速容器。该工具包包括一个容器运行时库和实用程序,用于自动配置容器以利用 NVIDIA GPU。
一、安装
1. 配置依赖库
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \&& curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
(可选)配置存储库以使用实验包:
sed -i -e '/experimental/ s/^#//g' /etc/apt/sources.list.d/nvidia-container-toolkit.list
2. 从存储库更新包列表
sudo apt-get update
3. 安装 NVIDIA Container Toolkit 软件包
sudo apt-get install -y nvidia-container-toolkit
二、配置Docker
1. 使用以下命令配置容器运行时nvidia-ctk
:
sudo nvidia-ctk runtime configure --runtime=docker
该nvidia-ctk
命令修改/etc/docker/daemon.json
主机上的文件。该文件已更新,以便 Docker 可以使用 NVIDIA 容器运行时。
2. 重新启动 Docker 守护进程:
sudo systemctl restart docker