# Переопределение photo-ai для работы на NVIDIA GPU.
# Использование:
#   docker compose -f docker-compose.yml -f docker-compose.gpu.yml up -d --build photo-ai
#
# Требуется драйвер NVIDIA и установленный NVIDIA Container Toolkit
# (nvidia-ctk + nvidia-container-runtime). Без toolkit контейнер не стартует —
# базовый docker-compose.yml остаётся CPU-сборкой и этот файл не подключается.
# PHOTO_AI_DEVICE=cuda при недоступной CUDA не роняет сервис: app.py пишет WARN
# и работает на CPU, /health при этом отвечает 200.
services:
  photo-ai:
    build:
      args:
        TORCH_VARIANT: cu124
    environment:
      PHOTO_AI_DEVICE: ${PHOTO_AI_DEVICE:-cuda}
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]