roop-unleashed / compose.gpu.yaml
cedrus
Add GPU Docker runtime config
aacdea2
Raw
History Blame Contribute Delete
582 Bytes
services:
roop-unleashed:
image: roop-unleashed:gpu
container_name: roop-unleashed
build:
context: .
dockerfile: Dockerfile.gpu
platform: linux/amd64
gpus: all
environment:
NVIDIA_VISIBLE_DEVICES: all
NVIDIA_DRIVER_CAPABILITIES: compute,utility
ports:
- "7860:7860"
volumes:
- ./output:/app/output
- ./models:/app/models
- ./models/insightface:/root/.insightface/models
- ./temp:/app/temp
- ./config.gpu.yaml:/app/config.yaml
command: python run.py
stdin_open: true
tty: true