トップページ -> インストール,設定,利用 -> Docker -> NVIDIA コンテナ・ツールキット(Ubuntu 上)
[サイトマップへ], [サイト内検索へ]

NVIDIA コンテナ・ツールキット(Ubuntu 上)

NVIDIA コンテナ・ツールキットは,NVIDIA CUDA をコンテナで動かすときに便利.

前準備

NVIDIA グラフィックスボードの確認とドライバのインストール

参考Webページ https://docs.nvidia.com/deploy/cuda-compatibility/index.html

グラフィックスボードの確認

CUDA 対応の GPU であるかを確認のため, 端末で,次のコマンドを実行.

lspci | grep -i nvidia | grep VGA

実行結果の例.(表示が空になるときは,NVIDIAのグラフィックスボードが無い可能性がある).

X.org の設定の確認

glxinfo | grep direct

[image]

xdriinfo

[image]

NVIDIA ドライバのインストール

NVIDIA グラフィックスボードがある場合に限り,ドライバのインストールを行う

  1. 端末で,次のコマンドを実行.
    sudo ubuntu-drivers autoinstall
    

  2. ドライバのインストール後は,システムの再起動
    sudo shutdown -r now
    
  3. 確認

    システムの再起動後,端末で,次のコマンドを実行し,ドライバ等を確認する

    先頭の「Driver Version」のところに,ドライバのバージョンが表示される.NVIDIA CUDA がインストール済みのときは,CUDA のバージョンも表示される.

    nvidia-smi
    


NVIDIA コンテナ・ツールキットのインストール

https://github.com/NVIDIA/nvidia-dockerに記載の手順に従う

  1. Docker のインストール
    sudo apt -yV install containerd docker.io docker-ce docker-ce-cli
    

    [image]
  2. apt パッケージリストの更新
    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
    

    ※ エラーが出る場合には,次のようなコマンドを試してみる.「18.04」のところは Ubuntu のバージョンにあわせて読み替えること.

    curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
    curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
    

    [image]
  3. インストール
    sudo apt -y update
    sudo apt -yV install nvidia-container-toolkit
    sudo systemctl restart docker
    

NVIDIA コンテナ・ツールキットを使ってみる

docker run -it --rm --gpus all nvidia/cuda:10.1-base /bin/bash
nvidia-smi
exit

[image]
docker run -it --rm --gpus all nvidia/cuda /bin/bash
nvidia-smi
exit

[image]

本サイトは金子邦彦研究室のWebページです.サイトマップは,サイトマップのページをご覧下さい. 本サイト内の検索は,サイト内検索のページをご利用下さい.

問い合わせ先: 金子邦彦(かねこ くにひこ) [image]