services:
  backend:
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    environment:
      MODEL_DEVICE: cuda
      NVIDIA_VISIBLE_DEVICES: all