docker-compose-nvidia.yaml

version: '3.8'

services:
  comfyapi-nvidia:
    build:
      context: .
      dockerfile: deployment/Dockerfile.local
      args:
        - GPU_TYPE=CUDA  # Set to CUDA for NVIDIA GPUs
    image: comfyapi-nvidia
    container_name: comfyapi-nvidia
    ports:
      - "${APP_LISTEN_PORT:-8000}:${APP_LISTEN_PORT:-8000}"
      - "${COMFYUI_LISTEN_PORT:-8001}:${COMFYUI_LISTEN_PORT:-8001}"
    gpus: all
    env_file:
      - .env
    environment:
      NVIDIA_VISIBLE_DEVICES: all
      NVIDIA_DRIVER_CAPABILITIES: compute,utility
      COMFYUI_INSTALL_PATH: '/app/ComfyUI'
      COMFYUI_WORKSPACE_PATH: '/app/comfyui_workspace'
    volumes:
      - ${COMFYUI_WORKSPACE_PATH}:/app/comfyui_workspace
    healthcheck:
      test: [ "CMD-SHELL", "wget --spider --quiet http://${APP_LISTEN_ADDRESS:-0.0.0.0}:${APP_LISTEN_PORT:-8000}/docs || exit 1" ]
      interval: 30s
      timeout: 30s
      retries: 3
      start_period: 5s
    command: uvicorn src.api.app:app --host ${APP_LISTEN_ADDRESS:-0.0.0.0} --port ${APP_LISTEN_PORT:-8000}