docker-compose.yml

services:
  DeepLiveCam:
    container_name: ${CONTAINER_NAME}
    image: ${IMAGE_NAME}
    build:
      context: .
      dockerfile: ${DOCKERFILE_PATH}
    ports:
      - "${PORT}:${CONTAINER_PORT}"
    environment:
      # System environments
      - NVIDIA_VISIBLE_DEVICES=${NVIDIA_VISIBLE_DEVICES}
      - DISPLAY=${DISPLAY}
    volumes:
      - ${APP_DIR}:/app
      - ${STARTUP_SCRIPT}:/app/docker_script.sh
      - ${MODELS_DIR}/.insightface:/root/.insightface
      - ${MODELS_DIR}/.insightface:/app/.insightface
      - ${MODELS_DIR}/gfpgan:/app/gfpgan
      - ${MODELS_DIR}:/app/models
      - ${OUTPUT_DIR}:/app/output
      - /tmp/.X11-unix:/tmp/.X11-unix
      - /dev/video0:/dev/video0
      - /dev/video1:/dev/video1
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: ${GPU_COUNT}
              capabilities: [gpu]
    tty: true
    privileged: true
    runtime: nvidia
    command: [ "/bin/bash", "/app/docker_script.sh" ]