-
Notifications
You must be signed in to change notification settings - Fork 1
/
docker-compose.yml
62 lines (60 loc) · 1.74 KB
/
docker-compose.yml
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
version: "2.3"
services:
# tensorflow:
# image: tensorflow_serving-gpu:0.0.0
# runtime: nvidia
# environment:
# CUDA_VISIBLE_DEVICES: 0
# ports:
# - "8600:8600"
# - "8601:8601"
# command:
# - --per_process_gpu_memory_fraction=0.5
# hostname: serving-tensorflow
# container_name: serving-tensorflow
# networks:
# - serving-network
# volumes:
# - /etc/localtime:/etc/localtime
# labels:
# - deploy.author=bobo
# -
triton:
image: triton_serving-gpu:0.0.0
runtime: nvidia
environment:
CUDA_VISIBLE_DEVICES: 0
ports:
- "8000:8000"
- "8001:8001"
command:
- --per_process_gpu_memory_fraction=0.5
hostname: serving-triton
container_name: serving-triton
networks:
- serving-network
volumes:
- /etc/localtime:/etc/localtime
labels:
- deploy.author=bobo
ai:
image: algorithms_serving:0.0.0
environment:
# TF_SERVING_HOST: "serving-tensorflow:8600"
TRITON_SERVING_HOST: "serving-triton:8001"
AI_SERVING_VERSION: 0.0.0
ports:
- "8211:8211"
- "8210:8210"
hostname: django_serving
container_name: django_serving
networks:
- serving-network
volumes:
- /etc/localtime:/etc/localtime
- ./image/:/root/dist/image
mac_address: 8a:ca:58:b9:e9:51
labels:
- deploy.author=bobo
networks:
serving-network: