NVIDIA Container Toolkit(NVIDIA Docker)

2024-01-09 18:00:56

引言

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

文章来源:https://blog.csdn.net/weixin_54703767/article/details/135479926
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。