-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathdocker-compose.gpu.yml
More file actions
140 lines (131 loc) · 3.1 KB
/
docker-compose.gpu.yml
File metadata and controls
140 lines (131 loc) · 3.1 KB
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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
version: "2.0"
services:
qdrant:
image: qdrant/qdrant:v1.1.2
ports:
- "127.0.0.1:6333:6333"
volumes:
- type: bind
source: ./data/qdrant
target: /qdrant/storage
indexer:
image: iart/indexer
build:
context: ./indexer
network_mode: "host"
volumes:
- type: bind
source: ./data/
target: /data/
- type: bind
source: ./indexer/
target: /indexer/
working_dir: /indexer/src/iart_indexer
environment:
- POETRY_VIRTUALENVS_CREATE=False
- PYTHONPATH=/indexer/src/
command: python3 -m iart_indexer --mode server -v -c /indexer/config.gpu.json
redisai:
image: redislabs/redisai:1.2.7-gpu-bionic
ports:
- "127.0.0.1:6379:6379"
deploy:
resources:
reservations:
devices:
- capabilities: [ gpu ]
elasticsearch:
image: elasticsearch:7.9.1
volumes:
- type: bind
source: ./data/elasticsearch/
target: /usr/share/elasticsearch/data
environment:
- xpack.security.enabled=false
- discovery.type=single-node
ulimits:
memlock:
soft: -1
hard: -1
nofile:
soft: 65530
hard: 65530
ports:
- "127.0.0.1:9300:9300"
- "127.0.0.1:9200:9200"
postgres:
image: postgres:13.2
volumes:
- ./data/db/:/var/lib/postgresql/data
environment:
- POSTGRES_DB=iart
- POSTGRES_USER=postgres
- POSTGRES_PASSWORD=postgres
- PGDATA=/var/lib/postgresql/data/data/
ports:
- "127.0.0.1:5432:5432"
memcached:
image: memcached:1.6.10-buster
command: -I 32m
ports:
- "127.0.0.1:11211:11211"
backend:
image: iart/backend
build:
context: ./backend
network_mode: "host"
volumes:
- type: bind
source: ./data/
target: /data
# Development mount
- type: bind
source: ./backend/src/iart_backend/
target: /app/iart_backend/
- type: bind
source: /home/springsteinm/sshfs/
target: /mnt/
working_dir: /app/iart_backend
environment:
- POETRY_VIRTUALENVS_CREATE=False
- PYTHONPATH=/app/
command: python3 manage.py runserver 0.0.0.0:8000
depends_on:
- postgres
- memcached
celery:
image: iart/backend
build:
context: ./backend
network_mode: "host"
volumes:
- type: bind
source: ./data/
target: /data
# Development mount
- type: bind
source: ./backend/src/iart_backend/
target: /app/iart_backend/
working_dir: /app/iart_backend
environment:
- GRPC_ENABLE_FORK_SUPPORT=1
- GRPC_POLL_STRATEGY=poll
- POETRY_VIRTUALENVS_CREATE=False
- PYTHONPATH=/app/
command: celery -A iart worker -l INFO
depends_on:
- postgres
- memcached
frontend:
image: iart/frontend
build:
context: ./frontend
network_mode: "host"
volumes:
- type: bind
source: ./frontend/
target: /frontend
- type: bind
source: ./data/
target: /data
working_dir: /frontend