前言

NVIDIA Container Toolkit 可用于各种 Linux 发行版并支持不同的容器引擎。在开始之前，请确保您已经为您的 Linux 发行版安装了 NVIDIA 驱动程序。安装驱动程序的推荐方法是使用分发包管理器，但也可以使用其他安装程序机制（例如，通过.run从 NVIDIA Driver Downloads下载安装程序）

一、基本概念

Nvidia Container Toolkit,该工具使Docker的容器能与主机的Nvidia显卡进行interact。自 NVIDIAContainer Toolkit 1.7.0起，Ubuntu 18.04、Ubuntu 20.04 和 Ubuntu 22.04发行版包含对 Jetson 平台的支持。这意味着为这些发行版提供的安装说明有望在 Jetson 设备上运行。nvidia-docker2>= 2.8.0

二、操作步骤

1.添加源

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
   && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
   && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list \
   && curl -s -L https://nvidia.github.io/nvidia-container-runtime/experimental/$distribution/nvidia-container-runtime.list | sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list

2.安装重启

sudo apt-get update
sudo apt-get install -y nvidia-docker2
sudo systemctl restart docker

运行nvidia/cuda:10.0-base，如果本地没有会从网上拉取nvidia/cuda:10.0-base镜像。

sudo docker run --rm --gpus all nvidia/cuda:10.0-base nvidia-smi

表示成功。

总结

本文介绍Ubuntu安装Nvidia Container Toolkit的流程。