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