この記事では2021年9月時点での、Ubuntu 20.04にNVIDIA-dockerをインストールする方法を紹介します。
実は、2020年5月にもUbuntu20.04にNVIDIA-dockerをインストールする方法を紹介していましたが、情報が古いです。
現在はインストールが非常に簡単になったので、改めてインストール方法を紹介します。
↓過去のインストール方法です。
Ubuntu 20.04にNVIDIA Container ToolkitをインストールしてDockerfileを作成目次 非表示
NVIDIA-dockerのインストール
まず、DockerとNVIDIA-dockerをインストールしていきます。
下記のコマンドは2021年9月時点で有効なコマンドです。
これらはNVIDIAの公式ドキュメントを参考にしたものなので、将来的に更新が入る可能性があります。
英語が苦手でなければ、実行前に公式ドキュメントを一度確認してみてください。
nvidia-dockerのGithub：https://github.com/NVIDIA/nvidia-docker
Installation Guide：https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker
dockerのインストール
ここはすでにDockerをインストール済みの方は必要ありません。
curl https://get.docker.com | sh \ && sudo systemctl --now enable dockerこのコマンドの詳細が気になる方は、dockerの公式ドキュメントを参考にしてください。
参考：Dockerのインストール：https://docs.docker.com/engine/install/ubuntu/#install-using-the-convenience-script
nvidia-dockerのインストール
動作確認を含めて、５回のコマンド実行でインストール可能です。
# GPGキーの取得
$ 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
$ sudo apt-get update
$ sudo apt-get install -y nvidia-docker2
$ sudo systemctl restart docker
# 動作確認
$ sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.57.02 Driver Version: 470.57.02 CUDA Version: 11.4 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 NVIDIA GeForce ... On | 00000000:01:00.0 Off | N/A |
| N/A 62C P3 N/A / N/A | 341MiB / 2002MiB | 0% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
+-----------------------------------------------------------------------------+まとめ
Ubuntu20.04にNVIDIA-dockerをインストールする方法を紹介しました。
動作確認も含めてたった５回のコマンド実行でインストール可能です。
非常に簡単になりましたね。これでDocker+GPUの開発も捗りそうです。
