Untitled

services:
  audiomuse-ai-worker01:
    container_name: audiomuse-nvidia-test-worker01
#    build:
#      context: ../..
#      dockerfile: Dockerfile
#      args:
#        BASE_IMAGE: nvidia/cuda:13.3.1-cudnn-runtime-ubuntu24.04
    image: ghcr.io/neptunehub/audiomuse-ai:latest-nvidia
    runtime: nvidia
    environment:
      SERVICE_TYPE: "worker"
      TZ: "${TZ:-UTC}"
      POSTGRES_USER: ${POSTGRES_USER:-youraudiomusepostgresuser}
      POSTGRES_PASSWORD: ${POSTGRES_PASSWORD:-youraudiomusepostgrespassword}
      POSTGRES_DB: ${POSTGRES_DB:-audiomusedb}
      POSTGRES_HOST: "${POSTGRES_HOST:-audiomuseIPhere}”
      POSTGRES_PORT: "${POSTGRES_PORT:-5432}"
      NVIDIA_VISIBLE_DEVICES: "0"
      NVIDIA_DRIVER_CAPABILITIES: "compute,utility"
      USE_GPU_CLUSTERING: "${USE_GPU_CLUSTERING:-true}"
      TEMP_DIR: "/app/temp_audio"
    volumes:
      - temp-audio-worker:/app/temp_audio
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: ["0"]
              capabilities: [gpu]
    restart: unless-stopped