docker-compose.yaml

# Docker Compose configuration for element-deeplabcut client environment
# Usage: docker compose up --build
#        docker compose run client python your_script.py

version: "3.8"

services:
  # MySQL database service
  db:
    restart: always
    image: datajoint/mysql:${MYSQL_VER:-8.0}
    environment:
      - MYSQL_ROOT_PASSWORD=${DJ_PASS:-datajoint}
    ports:
      - "${DB_PORT:-3306}:3306"
    volumes:
      - db_data:/var/lib/mysql
    healthcheck:
      test: ["CMD", "mysqladmin", "ping", "-h", "localhost", "-u", "root", "-p${DJ_PASS:-datajoint}"]
      timeout: 15s
      retries: 10
      interval: 15s
    networks:
      - element_network

  # Client container
  client:
    build:
      context: .
      dockerfile: Dockerfile
    depends_on:
      db:
        condition: service_healthy
    environment:
      - DJ_HOST=db
      - DJ_USER=root
      - DJ_PASS=${DJ_PASS:-datajoint}
      - DJ_PORT=3306
      - DATABASE_PREFIX=${DATABASE_PREFIX:-}
      - DLC_ROOT_DATA_DIR=${DLC_ROOT_DATA_DIR:-/app/test_videos}
      - PYTHONUNBUFFERED=1
      # GPU support - set NVIDIA_VISIBLE_DEVICES to enable GPU access
      # Can be overridden via -e flag in docker compose run
      - NVIDIA_VISIBLE_DEVICES=${NVIDIA_VISIBLE_DEVICES:-all}
    volumes:
      # Mount data directory (can be customized via DLC_DATA_DIR env var)
      # Default: ./test_videos (for tests) or ./data (for general use)
      - ${DLC_DATA_DIR:-./test_videos}:/app/data
      # Mount config file if it exists
      - ./dj_local_conf.json:/app/dj_local_conf.json:ro
      # Mount the project directory for development
      - .:/app
    networks:
      - element_network
    # GPU support for docker compose v2
    # This requires nvidia-container-toolkit to be installed on the host
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    # Default command - interactive bash (entrypoint is /bin/bash)
    # Example: docker compose run client -c "python test_model_train_run.py"
    command: ["-i"]
    stdin_open: true
    tty: true

volumes:
  db_data:

networks:
  element_network:
    driver: bridge