【Ubuntu 24.04】NVIDIA Container Toolkit（nvidia-docker）のインストール方法

以前までは、dockerでnvidiaのGPUを使うため、nvidia-docker / nvidia-docker2をインストールしていたと思います。

しかし、2024年1月にNVIDIA Container Toolkitに置き換わりました。

既存のnvidia-dockerは非推奨となり、リポジトリはアーカイブされました。

この記事では、新しくなったNVIDIA Container Toolkitについて、インストール方法をまとめます。

NVIDIA Container Toolkitとは

NVIDIA Container Toolkitとは、Dockerのコンテナ上でNVIDIAのGPUを利用できるようにするためのツールキットです。

GitHub - NVIDIA/nvidia-container-toolkit: Build and run containers leveraging NVIDIA GPUs

Build and run containers leveraging NVIDIA GPUs. Contribute to NVIDIA/nvidia-container-toolkit development by creating a...

github.com

以前まではnvidia-dockerでしたが、2024年1月以降はその後継として利用可能になっています。

NVIDIA Container Toolkitのインストール方法

NVIDIA Container Toolkitをインスールする前提条件

前提条件は以下の2つです。

• NVIDIAのGPUを搭載していること
• NVIDIA driverがインストールされていること
• dockerがインストールされていること

NVIDIA Container Toolkitのインストール手順

1. GPGキーの取得と保存をします。

curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg

2. NVIDIAのAPTリポジトリリストをダウンロードし、保存します。

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

3. APTパッケージリストを更新します。

sudo apt update

4. APTでNVIDIA Container Toolkitをインストールします。

sudo apt install nvidia-container-toolkit -y

5. dockerサービスに読み込ませるため、PCの再起動または以下のコマンドでサービスの再起動をします。

systemctl restart docker.service

NVIDIA Container Toolkitがインストールされていることの確認

以下のコマンドでNVIDIAのGPUを使用するDockerコンテナを起動し、nvidia-smiを実行することで、NVIDIA Container Toolkitがインストールされていることの確認を行います。

docker run --gpus all nvidia/cuda:12.4.1-runtime-ubuntu22.04 nvidia-smi

以下のように表示されていれば、正常な動作をしています。