https://docs.nvidia.com/datacenter/cloud-native/

5. GPU使用時の設定

高性能な開発環境を構築するため,NVIDIA GPU を使用する場合は,以下のコマンドで必要なツールキットをインストールする.

sudo apt install -y nvidia-container-toolkit
sudo systemctl restart docker
  • NVIDIA イメージからのコンテナの作成と開始

    前準備として,「sudo apt install -y nvidia-container-toolkit」によるインストールが必要.

    次に,https://hub.docker.com/r/nvidia/cuda/tagsでタグを確認.下のコマンドでは,タグとして「nvidia/cuda:12.8.0-cudnn-devel-ubuntu24.04」を使用

    docker run --name myubuntu --restart unless-stopped -p 8888:8888 -it --gpus all nvidia/cuda:12.8.0-cudnn-devel-ubuntu24.04 /bin/bash
    nvidia-smi
    exit