schuyler/docker-compose.yaml at main · HPI-Information-Systems/schuyler · 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
version: "3.8"
services:
  app:
    tty: true
    build:
      context: ../
      dockerfile: ./schuyler/dockerfile
    volumes:
      - ./result:/results
      - ../data:/data
      - /data/huggingface_cache/huggingface/hub/:/models
      - ./:/experiment
    command: >
      python3 /experiment/schuyler/experimenter/experiment_script.py --scenario base_experiment --tag new --wandb
    env_file:
      - ./.env
    depends_on:
      - db
    runtime: nvidia
    environment:
      PYTHONUNBUFFERED: 1
      CUDA_VISIBLE_DEVICES: "0"
      POSTGRES_HOST: db
      POSTGRES_PORT: 5432
      POSTGRES_DB: postgres
      POSTGRES_USER: user
      POSTGRES_PASSWORD: password
      PGPASSWORD: password

  db:
    image: postgres:14
    container_name: postgres_db_schuyler
    environment:
      POSTGRES_DB: postgres
      POSTGRES_USER: user
      POSTGRES_PASSWORD: password
    ports:
      - "5432:5432"
    volumes:
      - postgres_data:/var/lib/postgresql/data

volumes:
  postgres_data: