docker-composeからNVIDIAのGPUを使う [2021年5月版]
docker-composeからNVIDIAのGPUを使うためにいろいろ調べてみたところ、NVIDIA Docker2やら何やらいろんな情報が錯綜しているので現状を調べてみた記録です。
※この記事は2021年5月23日現在の情報です。最新情報については公式ドキュメント等を参照してください。
DockerでNVIDIAのGPUを使う方法
現時点ではDockerをインストールし、その後nvidia-docker2
をインストールするのが良さそうです。
Docker 19.03以降ではGPUをサポートするようになり、Dockerのインストール+NVIDIA Container Toolkitをインストールすることで、docker
コマンドのオプションからGPUを呼び出せるようになるようです。
また、nvidia-docker2
をインストールすればnvidia-container-toolkit
も自動でインストールされるようなので、現時点ではDockerをインストールし、その後nvidia-docker2
をインストールするのが良さそうです。
最も詳しいと思われる上記のNVIDIA Japan公式のMediumの記事がNVIDIAのGPUをDockerから使う際に参考になると思います。
docker-composeからNVIDIAのGPUを使う方法
- docker-compose 1.27までの場合
runtime: nvidia
を指定するruntime
を指定できるのはdocker-compose
の2.3以降のバージョンのようです*1
- docker-compose 1.28以降の場合
devices:
とcapabilities:
を指定する、またはruntime: nvidia
を指定する
ようです*2。
※Docker 19.03以降+NVIDIA Container Toolkitをインストールしていて、Rootlessモードを使っていない前提