forked from JeffreyCA/spleeter-web
-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
docker-compose.gpu.yml
102 lines (102 loc) · 2.67 KB
/
docker-compose.gpu.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
version: '3.4'
x-celery-env: &celery-env
- DJANGO_SETTINGS_MODULE=django_react.settings_docker
- CELERY_BROKER_URL=redis://redis:6379/0
- CELERY_RESULT_BACKEND=redis://redis:6379/0
- CPU_SEPARATION=0
- CELERY_FAST_QUEUE_CONCURRENCY=${CELERY_FAST_QUEUE_CONCURRENCY:-3}
- CELERY_SLOW_QUEUE_CONCURRENCY=${CELERY_SLOW_QUEUE_CONCURRENCY:-1}
- DEFAULT_FILE_STORAGE
- AWS_ACCESS_KEY_ID
- AWS_SECRET_ACCESS_KEY
- AWS_STORAGE_BUCKET_NAME
- AWS_S3_CUSTOM_DOMAIN
- AZURE_ACCOUNT_KEY
- AZURE_ACCOUNT_NAME
- AZURE_CONTAINER
- AZURE_CUSTOM_DOMAIN
- NVIDIA_VISIBLE_DEVICES=all
- NVIDIA_DRIVER_CAPABILITIES=all
services:
redis:
image: redis:6.0-buster
hostname: redis
expose:
- "6379"
volumes:
- redis-data:/data
restart: always
nginx:
image: jeffreyca/spleeter-web-nginx:${TAG:-latest}
volumes:
- staticfiles:/webapp/staticfiles
environment:
- API_HOST=api
depends_on:
- api
restart: always
api:
image: jeffreyca/spleeter-web-backend:${TAG:-latest}-gpu
volumes:
- assets:/webapp/frontend/assets
- sqlite-data:/webapp/sqlite
- staticfiles:/webapp/staticfiles
stdin_open: true
tty: true
environment:
- DJANGO_SETTINGS_MODULE=django_react.settings_docker
- CELERY_BROKER_URL=redis://redis:6379/0
- CELERY_RESULT_BACKEND=redis://redis:6379/0
- CPU_SEPARATION=0
- APP_HOST
- DEFAULT_FILE_STORAGE
- AWS_ACCESS_KEY_ID
- AWS_SECRET_ACCESS_KEY
- AWS_STORAGE_BUCKET_NAME
- AWS_S3_CUSTOM_DOMAIN
- AZURE_ACCOUNT_KEY
- AZURE_ACCOUNT_NAME
- AZURE_CONTAINER
- AZURE_CUSTOM_DOMAIN
- YOUTUBE_API_KEY
depends_on:
- redis
- frontend
restart: always
celery-fast:
image: jeffreyca/spleeter-web-backend:${TAG:-latest}-gpu
entrypoint: ./celery-fast-entrypoint.sh
volumes:
- celery-data:/webapp/celery
- pretrained-models:/webapp/pretrained_models
- sqlite-data:/webapp/sqlite
runtime: nvidia
environment: *celery-env
depends_on:
- redis
restart: always
celery-slow:
image: jeffreyca/spleeter-web-backend:${TAG:-latest}-gpu
entrypoint: ./celery-slow-entrypoint.sh
volumes:
- celery-data:/webapp/celery
- pretrained-models:/webapp/pretrained_models
- sqlite-data:/webapp/sqlite
runtime: nvidia
environment: *celery-env
depends_on:
- redis
restart: always
frontend:
image: jeffreyca/spleeter-web-frontend:${TAG:-latest}
volumes:
- assets:/webapp/frontend/assets
stdin_open: true
tty: true
volumes:
assets:
celery-data:
pretrained-models:
redis-data:
sqlite-data:
staticfiles: