docker - 在 docker compose 上使用 GPU

Question

我有一个由 nginx 和烧瓶应用程序（运行 tensorflow）组成的 docker compose 文件。像这样的东西：

version: '3.7'

services:
    ...
    flask:
      expose:
        - 6006
#      deploy:
#        resources:
#          limits:
#            gpus: all

    nginx:
      ...
      restart: always
      ports:
        - "6006:80"
     ...

它工作正常，但我想为我的烧瓶应用程序使用 gpu 资源。我可以使用--gpu标志运行带有 gpu 的烧瓶应用程序（docker-container），但这与 nginx 容器是分开的。

目前，compose 3.7 不支持它。github中的问题。

问题：运行我的应用程序的最佳解决方法是什么？降级docker-compose？我可以将 nginx 和烧瓶组合在同一个容器中吗？

标签： dockertensorflownginxdocker-compose