36 lines
770 B
YAML
36 lines
770 B
YAML
services:
|
|
skin-py-analysis:
|
|
build: /data/www/pifu
|
|
image: skin-py-gpu:1.0.0
|
|
working_dir: /app
|
|
ports:
|
|
- "8000:8000"
|
|
volumes:
|
|
- /data/www/pifu:/app
|
|
- /data/analysis:/data/analysis
|
|
environment:
|
|
TZ: Asia/Shanghai
|
|
PYTHONUNBUFFERED: 1
|
|
TASK_DB_PATH: /data/analysis/.worker/tasks.sqlite3
|
|
MAX_CONCURRENT_TASKS: 6
|
|
MAX_PENDING_TASKS: 100
|
|
MODEL_PATH: /app/best.pt
|
|
IMAGE_WORKERS: 4
|
|
|
|
NVIDIA_VISIBLE_DEVICES: all
|
|
NVIDIA_DRIVER_CAPABILITIES: compute,utility
|
|
command: python api_server.py
|
|
restart: unless-stopped
|
|
gpus: all
|
|
networks:
|
|
- default
|
|
- common-network
|
|
|
|
networks:
|
|
default:
|
|
external: true
|
|
name: skin-network
|
|
|
|
common-network:
|
|
external: true
|