【トラブルシューティング】Docker composeでNVIDIA GPUが利用できなくなった時の対処方法

はじめに

　DockerとDocker Composeをアップデート後に、NVIDIA GPUが利用できなくなった時の対処方法を紹介する。

動作環境

• Ubuntu 22.04 x64

原因と対策

　docker-compose.ymlの設定ミス。利用するGPU数（count）を設定しなかったのが原因。以前はcountを設定していなくてもNVIDIA GPUが使えていたが、アップデート後は使えなくなった（2024/9/25時点）。
　countを以下のように設定することでNVIDIA GPUが認識するようになる。

docker-compose.yml

services:
 test:
   image: nvidia/cuda:10.2-base
   command: nvidia-smi
   deploy:
     resources:
       reservations:
         devices:
           - driver: nvidia
             count: 1  # この設定を忘れないようにする。全てのGPUを利用する場合は、all にする。
             capabilities: [gpu]

まとめ

　Docker composeでNVIDIA GPUが利用できなくなった時の対処方法を紹介した。設定方法が変わる可能性があるので、公式ドキュメントを参照することをお勧めする。

参考