1. 環境
CentOS8
Nvidia Driver 430.14
Docker 19.03.8
2. Dockerインストール
こちらのページを参考にインストールを行った。
https://qiita.com/cyberblack28/items/0b0ec02bce67a16e2f17
以前は nvidia-docker コマンドとかがあったみたいだが、最近は docker コマンドからGPUが使えるようになったみたい
https://qiita.com/ksasaki/items/b20a785e1a0f610efa08
特に過去のしがらみ等はないので、「Docker 19.03 + nvidia-container-toolkit」のインストールを行った。
3. nvidia-container-toolkitのインストール
以下のQuickStartにはCentOS8の記載はないが、CentOS7と同じ方法でインストール可能
4. 動作確認
$ docker run --gpus 1 --rm nvidia/cuda nvidia-smi
Sun May 3 03:53:16 2020
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 430.14 Driver Version: 430.14 CUDA Version: 10.2 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX 105... Off | 00000000:07:00.0 Off | N/A |
| 29% 38C P8 N/A / 75W | 0MiB / 4040MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| No running processes found |
+-----------------------------------------------------------------------------+
これでDockerからGPUを利用できそう
以上