NVIDIA Container Toolkit を使って Docker のコンテナからホストの GPU を使ってみるという内容です。Docker 19.03 から追加されたオプション --gpus を使って GPU コンテナを作成します。 環境OS は Ubuntu 18.04。NVIDIA ドライバーはインストール済みであり、Docker も利用可能な状態です。 $ cat /etc/lsb-release DISTRIB_ID=Ubuntu DISTRIB_RELEASE=18.04 DISTRIB_CODENAME=bionic DISTRIB_DESCRIPTION="Ubuntu 18.04.3 LTS" $ nvidia-smi Fri Nov 29 19:23:30 2019 +------------------------------------------------
![DockerコンテナでもGPUが使いたい](https://cdn-ak-scissors.b.st-hatena.com/image/square/85c7098233ebb711c545ba84a7a3366cb261eeff/height=288;version=1;width=512/https%3A%2F%2Fblog.mahoroi.com%2Fstatic%2F94c582dac90e6f15aea56d9966ad0b34%2F5a190%2Fogp.png)