28 lines
625 B
YAML
28 lines
625 B
YAML
services:
|
|
api:
|
|
build:
|
|
context: .
|
|
args:
|
|
VARIANT: gpu
|
|
ports:
|
|
- "8000:8000"
|
|
volumes:
|
|
- ./data/uploads:/data/uploads
|
|
- ./data/outputs:/data/outputs
|
|
- ./data/models:/data/models
|
|
- ./data/temp:/data/temp
|
|
- ./data/jobs:/data/jobs
|
|
environment:
|
|
- RSR_EXECUTION_PROVIDERS=["cuda"]
|
|
- RSR_EXECUTION_THREAD_COUNT=8
|
|
- RSR_TILE_SIZE=400
|
|
- RSR_LOG_LEVEL=info
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: 1
|
|
capabilities: [gpu]
|
|
restart: unless-stopped
|