docker-compose.yml

deploy:
  resources:
    reservations:
      devices:
        - capabilities: [gpu]