docker - 如何使用 docker compose 指定 GPU 限制 docker swarm？

Question

如何使用 docker compose 指定 GPU、CPU 和内存限制？

对于这种情况，我已按照（https://nvidia.github.io/nvidia-container-runtime/）的说明进行操作。

使用 ：

$ apt-get install nvidia-container-runtime

我也在检查它：

$ docker run -it --rm --gpus all ubuntu nvidia-smi

例子docker-compose.yaml：

version: '2'
   ...
   services:
        ...
        my-service:
            ...
            deploy:
              resources:
                reservations:
                  generic_resources:
                    - discrete_resource_spec:
                        kind: 'gpu'
                        value: 1
                limits:
                      cpus: '4'
                      memory: 4096M
         another-service:
             ...
             deploy:
                resources:
                   limits:
                      cpus: '0.001'
                      memory: 50M

标签： dockerdocker-composedocker-swarm