File size: 505 Bytes
77834ec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
version: '3.8'
services:
segmentation-app:
build: .
ports:
- "7861:7860"
volumes:
- models_cache:/root/.cache
- ./outputs:/app/outputs # Map host outputs directory to container outputs
environment:
- NVIDIA_VISIBLE_DEVICES=all
- NVIDIA_DRIVER_CAPABILITIES=compute,utility
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [gpu]
volumes:
models_cache: |