Junkie Work

access: 264489

NVIDIA Container Toolkitをインストール


NVIDIA Container Toolkitをインストールします。NVIDIA Container Toolkitは、NVIDIA GPUをコンテナで利用するためのツールです。

前提パッケージを追加

$ sudo apt update 
$ sudo apt install --no-install-recommends  ca-certificates  curl  gnupg2

リポジトリの追加

$ 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
$ sudo apt update

NVIDIA Container Toolkitをインストール

$ sudo apt install nvidia-container-toolkit

Dockerへ設定

$ sudo nvidia-ctk runtime configure --runtime=docker
$ sudo systemctl restart docker

動作確認

以下のコマンドを実行して、NVIDIA Container Toolkitが正しくインストールされ、DockerでNVIDIA GPUが使用できるか確認します。

$ sudo docker run --rm --gpus all nvidia/cuda:13.4.2-base-ubuntu26.04 nvidia-smi

Docker HubのNVIDIA CUDA Docker Imagesのページには、CUDAのバージョンごとに様々なDockerイメージが用意されています。必要に応じて、適切なイメージを選択して使用してください。

NVIDIA CUDA Docker Imagesでbase-ubuntuを検索