Initial Real-ESRGAN API project setup
This commit is contained in:
27
docker-compose.gpu.yml
Normal file
27
docker-compose.gpu.yml
Normal file
@@ -0,0 +1,27 @@
|
||||
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
|
||||
Reference in New Issue
Block a user