gpu-docker-api/docker-compose.yaml at main · XShengTech/gpu-docker-api · 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
version: '3.5'

networks:
  etcd-net:
    name: etcd-net

volumes:
  etcd-data:
    name: etcd-data
  gpu-docker-api-data:
    name: gpu-docker-api-data

services:
  etcd:
    hostname: etcd
    image: bitnami/etcd:3.4.28
    container_name: etcd
    restart: unless-stopped
    networks:
      - etcd-net
    environment:
      - "ETCD_ADVERTISE_CLIENT_URLS=http://0.0.0.0:2379"
      - "ETCD_LISTEN_CLIENT_URLS=http://0.0.0.0:2379"
      - "ETCD_LISTEN_PEER_URLS=http://0.0.0.0:2380"
      - "ETCD_INITIAL_ADVERTISE_PEER_URLS=http://0.0.0.0:2380"
      - "ALLOW_NONE_AUTHENTICATION=yes"
      - "ETCD_INITIAL_CLUSTER=node1=http://0.0.0.0:2380"
      - "ETCD_NAME=node1"
      - "ETCD_DATA_DIR=/opt/bitnami/etcd/data"
    ports:
      - "2379:2379"
      - "2380:2380"
    volumes:
      - etcd-data:/bitnami/etcd/data

  gpu-docker-api:
    hostname: gpu-docker-api
    image: xshengtech/gpu-docker-api:latest
    container_name: gpu-docker-api
    restart: unless-stopped
    network_mode: host
    runtime: nvidia
    environment:
      - APIKEY=CHANGETHIS
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [ gpu ]
    volumes:
      - /var/run/docker.sock:/var/run/docker.sock
      - /etc/localtime:/etc/localtime:ro
      - PATH_TO_DOCKER_STORAGE:PATH_TO_DOCKER_STORAGE
      - gpu-docker-api-data:/data/merges