# GPU configuration (NVIDIA CUDA)
# Requires: nvidia-container-toolkit installed on host
# Install: https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html

services:
  server:
    build:
      context: .
      dockerfile: Dockerfile
    image: uncloseai-speech:local
    env_file: speech.env
    ports:
      - "8000:8000"
    volumes:
      - ./voices:/app/voices
      - ./config:/app/config
      - ./cloned-voices:/app/cloned-voices
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
    restart: unless-stopped
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              # Uncomment to select specific GPU(s):
              # device_ids: ['0']
              count: all
              capabilities: [gpu]
