version: "3.8"

networks:
  argus-sys-net:
    external: true

services:
  metric-gpu-node:
    image: ${NODE_GPU_BUNDLE_IMAGE_TAG:-argus-sys-metric-test-node-bundle-gpu:latest}
    container_name: argus-metric-gpu-node-swarm
    hostname: ${GPU_NODE_HOSTNAME:-swarm-metric-gpu-001}
    restart: unless-stopped
    privileged: true
    runtime: nvidia
    environment:
      - TZ=Asia/Shanghai
      - DEBIAN_FRONTEND=noninteractive
      - MASTER_ENDPOINT=${MASTER_ENDPOINT:-http://master.argus.com:3000}
      - ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
      - ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
      - AGENT_ENV=${AGENT_ENV:-dev2}
      - AGENT_USER=${AGENT_USER:-yuyr}
      - AGENT_INSTANCE=${AGENT_INSTANCE:-gpu001sX}
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
      - GPU_MODE=gpu
    networks:
      argus-sys-net:
        aliases:
          - ${AGENT_INSTANCE}.node.argus.com
    volumes:
      - ./private-gpu-nodes/argus/agent:/private/argus/agent
    command: ["sleep", "infinity"]