cudascope/docker-stack.yml at main · ssubbotin/cudascope · 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
version: "3.8"

services:
  # Agent: runs on every GPU node, collects metrics and pushes to hub
  agent:
    image: ssubbotin/cudascope:latest
    command: ["--mode=agent", "--hub-url=http://hub:9090"]
    deploy:
      mode: global
      placement:
        constraints:
          - node.labels.gpu == true
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
        limits:
          memory: 128M
    networks:
      - cudascope

  # Hub: central server with UI and storage
  hub:
    image: ssubbotin/cudascope:latest
    command: ["--mode=hub"]
    ports:
      - "9090:9090"
    volumes:
      - cudascope-data:/data
    deploy:
      replicas: 1
      placement:
        constraints:
          - node.role == manager
    environment:
      - CUDASCOPE_DATA_DIR=/data
    networks:
      - cudascope

volumes:
  cudascope-data:

networks:
  cudascope:
    driver: overlay