Docker 组成相当于“docker run --gpu=all”选项

要自动化配置（docker run参数）用于启动 docker 容器，我正在写一个docker-compose.yml file.

我的容器应该可以访问 GPU，所以我目前使用docker run --gpus=all范围。这在公开 GPU 以供使用 https://docs.docker.com/config/containers/resource_constraints/ docs:

包括--gpus启动容器访问 GPU 时的标志 资源。指定要使用的 GPU 数量。例如：

$ docker run -it --rm --gpus 所有 ubuntu nvidia-smi

很遗憾，使用 Compose 启用 GPU 访问 https://docs.docker.com/compose/gpu-support/没有准确描述这个用例。本指南使用deployyaml 元素，但在保留带有 GPU 的机器的情况下。实际上，另一个文档 https://docs.docker.com/compose/compose-file/compose-file-v3/#deploy说它将被忽略docker-compose:

这仅在使用 docker stack deploy 部署到 swarm 时生效，并且被忽略docker-compose up 和 docker-compose run。

在尝试并解决了无数问题之后，我意识到这只是过时的文档。

将以下 yaml 块添加到我的docker-compose.yml导致nvidia-smi可供使用。

    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]