version: "3.8"

services:
  deepstream:
    image: nvcr.io/nvidia/deepstream:${DS_VERSION:-7.1}-triton-multiarch
    container_name: deepstream
    restart: unless-stopped
    ports:
      - "${RTSP_PORT:-8554}:8554"
      - "${REST_PORT:-9000}:9000"
    volumes:
      - deepstream_apps:/opt/nvidia/deepstream/deepstream/sources/apps
      - deepstream_models:/opt/nvidia/deepstream/deepstream/samples/models
      - deepstream_configs:/opt/nvidia/deepstream/deepstream/samples/configs
      - deepstream_streams:/opt/nvidia/deepstream/deepstream/samples/streams
    environment:
      - CUDA_VISIBLE_DEVICES=${CUDA_DEVICES:-all}
      - DISPLAY=${DISPLAY:-}
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu, video, compute, utility]
    runtime: nvidia
    network_mode: ${NETWORK_MODE:-bridge}
    shm_size: ${SHM_SIZE:-2g}
    # Required for video device access on edge nodes
    privileged: ${PRIVILEGED:-false}
    devices:
      - /dev/video0:/dev/video0

volumes:
  deepstream_apps:
  deepstream_models:
  deepstream_configs:
  deepstream_streams: