docker-compose で GPU を使う

docker run では --gpus all としていたのですが、docker-compose ではどう指定すればいいのかわからなかったので調べてみました。

基本的には以下に書かれている通りです。

• https://docs.docker.com/compose/gpu-support/

docker-compose では例えば以下のように記述します。

deploy:
    resources:
        reservations:
            devices:
                - capabilities: [gpu]
environment: 
    - NVIDIA_VISIBLE_DEVICES=all
    - NVIDIA_DRIVER_CAPABILITIES=all

参考 URL

• https://docs.docker.com/compose/gpu-support/
• https://stackoverflow.com/questions/66298638/devices-properties-is-not-allowed-while-creating-docker-compose-with-nvidia
• https://qiita.com/1024chon/items/4aaf4c1493fc9200cadc