docker-compose.yml

services:
  athena:
    build:
      context: .
      dockerfile: Dockerfile
    container_name: athena
    depends_on:
      postgres:
        condition: service_healthy
    command: ./scripts/run_experiments.py --models ours --workloads JOB --methods ours
    volumes:
      - ./data:/data
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
      - CUBLAS_WORKSPACE_CONFIG=:4096:8
    runtime: nvidia
    networks:
      - athena

  postgres:
    build:
      context: ./postgres_service
      dockerfile: Dockerfile
    container_name: postgres
    environment:
      - POSTGRES_USER=postgres
      - POSTGRES_PASSWORD=postgres
    volumes:
      - ./postgres_service/data:/var/lib/postgresql/data
    networks:
      - athena
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U postgres -h localhost"]
      interval: 1s
      timeout: 30s
      retries: 30

networks:
  athena:
    driver: bridge