ml-workflow-engine/docker-compose.yml at main · Sebastijan-Dominis/ml-workflow-engine · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
services:
  ml-workflow:
    build:
      context: .
    tty: true
    stdin_open: true
    # GPU support
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [ gpu ]
    volumes:
      - ./configs:/app/configs
      - ./data:/app/data
      - ./feature_store:/app/feature_store
      - ./experiments:/app/experiments
      - ./failure_management:/app/failure_management
      - ./orchestration_logs:/app/orchestration_logs
      - ./model_registry:/app/model_registry
      - ./predictions:/app/predictions
      - ./monitoring:/app/monitoring
      - ./docs:/app/docs
      - ./assets:/app/assets
    ports:
      - "8000:8000"
      - "8050:8050"
    env_file:
      - .env
    restart: unless-stopped