OpenAlphaDiffract/compose.yaml at main · AdvancedPhotonSource/OpenAlphaDiffract · 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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
services:
  downloader:
    build:
      context: .
      dockerfile: docker/downloader.Dockerfile
    env_file:
      - .env
    volumes:
      - ./data/:/data/
      - ./configs:/configs:ro
      - ./src/downloader:/app:ro
    user: "${UID}:${GID}"
    command: python /app/downloader.py --config /configs/download.yaml
    restart: "no"

  simulator:
    build:
      context: .
      dockerfile: docker/simulator.Dockerfile
    volumes:
      - ./data/:/data/
      - ./configs:/app/configs:ro
    environment:
      - MPLCONFIGDIR=/tmp/matplotlib
    user: "${UID}:${GID}"
    command: python -m simulator.diffraction_generator --config /app/configs/simulator.yaml
    restart: "no"

  trainer:
    build:
      context: .
      dockerfile: docker/trainer.Dockerfile
    volumes:
      - ./data/:/data/
      - ./configs:/configs:ro
      - ./outputs:/outputs
      - ./src/trainer:/app:ro
    environment:
      - PYTHONUNBUFFERED=1
      - MPLCONFIGDIR=/tmp/matplotlib
      - MKL_THREADING_LAYER=GNU
      - MKL_SERVICE_FORCE_INTEL=1
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
    runtime: nvidia
    user: "${UID}:${GID}"
    command: python /app/run_train_with_manifests.py /configs/trainer.docker.yaml
    restart: "no"

  ui:
    build:
      context: .
      dockerfile: docker/ui.Dockerfile
    ports:
      - "7860:7860"
    volumes:
      - ./data/:/data/:ro
      - ./src/ui/models/:/app/models/:ro
    environment:
      - PYTHONUNBUFFERED=1
      - PORT=7860
    restart: unless-stopped