Spaces:
Sleeping
Sleeping
| # GPU override file | |
| # Usage: docker compose -f docker-compose.yml -f docker-compose.gpu.yml up -d | |
| # | |
| # Prerequisites: | |
| # 1. NVIDIA GPU driver installed on host | |
| # 2. NVIDIA Container Toolkit installed | |
| # 3. Verify with: docker info | grep nvidia | |
| # | |
| # This file overrides the default docker-compose.yml to: | |
| # - Build with Dockerfile.gpu (NVIDIA CUDA base image) | |
| # - Attach GPU devices to both services because video generation can run in | |
| # either the WebUI process or the API process | |
| services: | |
| webui: | |
| build: | |
| context: . | |
| dockerfile: Dockerfile.gpu | |
| deploy: | |
| resources: | |
| reservations: | |
| devices: | |
| - driver: nvidia | |
| count: 1 | |
| capabilities: [gpu] | |
| api: | |
| build: | |
| context: . | |
| dockerfile: Dockerfile.gpu | |
| deploy: | |
| resources: | |
| reservations: | |
| devices: | |
| - driver: nvidia | |
| count: 1 | |
| capabilities: [gpu] | |