docker-compose-gpu.yml

version: '3.8'
services:
  video-matting:
    build: .
    restart: unless-stopped
    shm_size: 8g
    ports:
      - "7860:7860"
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              capabilities: [gpu]
              count: all