Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 17 additions & 3 deletions docker-compose.gpu.yml
Original file line number Diff line number Diff line change
Expand Up @@ -9,15 +9,20 @@ services:
image: marker-api-gpu-image
volumes:
- .:/app
- /etc/localtime:/etc/localtime:ro
- /etc/timezone:/etc/timezone:ro
depends_on:
- redis
environment:
- REDIS_HOST=${REDIS_HOST}
restart: unless-stopped
deploy:
resources:
reservations:
devices:
- capabilities: [gpu] # Request GPU support
- driver: nvidia
count: all
capabilities: [gpu] # Request GPU support

app:
container_name: marker-api-gpu
Expand All @@ -29,14 +34,19 @@ services:
- "8080:8080"
volumes:
- .:/app
- /etc/localtime:/etc/localtime:ro
- /etc/timezone:/etc/timezone:ro
depends_on:
- redis
- celery_worker
restart: unless-stopped
deploy:
resources:
reservations:
devices:
- capabilities: [gpu] # Request GPU support
- driver: nvidia
count: all
capabilities: [gpu] # Request GPU support

redis:
container_name: redis
Expand All @@ -52,6 +62,8 @@ services:
- 5556:5555
volumes:
- .:/app
- /etc/localtime:/etc/localtime:ro
- /etc/timezone:/etc/timezone:ro
environment:
- REDIS_HOST=${REDIS_HOST}
depends_on:
Expand All @@ -62,4 +74,6 @@ services:
resources:
reservations:
devices:
- capabilities: [gpu] # Request GPU support
- driver: nvidia
count: all
capabilities: [gpu] # Request GPU support