-
Notifications
You must be signed in to change notification settings - Fork 1
/
docker-compose.yml
104 lines (101 loc) · 2.81 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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
version: "3.7"
networks:
mt:
name: techiaith
services:
bombe-lab:
image: techiaith/marian-nmt-lab:0.1
runtime: nvidia
container_name: bombe-lab
build:
context: ./lab
args:
marian_compile_server: "${MARIAN_COMPILE_SERVER:-off}"
volumes:
- ./dictionaries:/dictionaries
- ./data/experiments:/experiments
- ./notebooks:/notebooks
- ./data/server-models:/server-models
- ./lab/src:/home/techiaith/app
- ./lab/supervisord.conf:/home/techiaith/supervisord.conf
entrypoint: ["supervisord", "-c", "/home/techiaith/supervisord.conf"]
ports:
- "8888:8888" # tensorboard graphs
- "9001:9001" # supervisorctrl
- "9010:9010" # jupyter notbook
restart: always
deploy:
resources:
reservations:
devices:
- driver: nvidia
capabilities: [gpu]
bombe-server_cy-en:
image: techiaith/marian-nmt-api:0.1
runtime: nvidia
container_name: bombe-api_cy-en
build:
context: ./server
args:
marian_compile_server: "${MARIAN_COMPILE_SERVER:-on}"
environment:
- "MARIAN_MODEL_NAME=${MARIAN_MODEL_NAME}"
- "MARIAN_WS_PORT=8080"
- "SOURCE_LANGUAGE=cy"
- "TARGET_LANGUAGE=en"
- "API_ALLOW_CORS_ORIGINS=${API_ALLOW_CORS_ORIGINS}"
volumes:
- /data/bombe/server-models:/models
- ./server/src:/home/techiaith/app
entrypoint: ['python', '-m', 'uvicorn',
'bombe.translation.api.views:app',
'--host', '0.0.0.0',
'--port', '8000']
ports:
- "8000:8000" # Fast API
- "8880:8080" # marian-server WebSocket
restart: always
deploy:
resources:
reservations:
devices:
- driver: nvidia
capabilities: [gpu]
bombe-server_en-cy:
image: techiaith/marian-nmt-api:0.1
runtime: nvidia
container_name: bombe-api_en-cy
build:
context: ./server
args:
marian_compile_server: "${MARIAN_COMPILE_SERVER:-on}"
environment:
- "MARIAN_MODEL_NAME=${MARIAN_MODEL_NAME}"
- "MARIAN_WS_PORT=8082"
- "SOURCE_LANGUAGE=en"
- "TARGET_LANGUAGE=cy"
- "API_ALLOW_CORS_ORIGINS=${API_ALLOW_CORS_ORIGINS}"
volumes:
- /data/bombe/server-models:/models
- ./server/src:/home/techiaith/app
entrypoint: ['python', '-m', 'uvicorn',
'bombe.translation.api.views:app',
'--host', '0.0.0.0',
'--port', '8002']
ports:
- "8002:8002" # Fast API
- "8882:8082" # marian-server WebSocket
restart: always
deploy:
resources:
reservations:
devices:
- driver: nvidia
capabilities: [gpu]
volumes:
dictionaries:
experiments:
notebooks:
server-config:
server-logs:
models: