| 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: |