SG-RAPL/compose.base.yaml at master · SashaDance/SG-RAPL · 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
63
64
65
66
67
services:
  llm_vicuna13b:
    build:
      context: .
      dockerfile: services/llm_vicuna13b/Dockerfile
    restart: always
    volumes:
      - $HOME/.cache/huggingface/hub/:/root/.cache/huggingface/hub/
      - ./common_packages:/common_packages
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: [ '0' ]
              capabilities: [ gpu ]
    networks:
      - assistant-network

  planner:
    build:
      context: .
      dockerfile: services/planner/Dockerfile
    restart: always
    volumes:
      - ./common_packages:/common_packages
    networks:
      - assistant-network

  actual_state:
    build:
      context: .
      dockerfile: services/actual_state/Dockerfile
    restart: always
    volumes:
      - ./common_packages:/common_packages
    networks:
      - assistant-network

  seg_and_track:
    build:
      context: .
      dockerfile: services/seg_and_track/Dockerfile
    volumes:
      - ./common_packages:/common_packages
      - seg_and_track_data:/data
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: [ '1' ]
              capabilities: [ gpu ]
    networks:
      - assistant-network

networks:
  assistant-network:
    driver: bridge

volumes:
  seg_and_track_data:
    driver: local
    driver_opts:
      o: bind
      type: none
      device: volumes/seg_and_track