YULI/pifu
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: use gpu

Browse Source
This commit is contained in:
zzw
2026-09-15 10:01:41 +08:00
parent a9d7aac95c
commit 9563d7bb98
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile
+1 -1
View File
@@ -1,4 +1,4 @@
FROM skin-py-cpu-base:v1.0.2 FROM skin-py-gpu:1.0.0
WORKDIR /app WORKDIR /app
docker-compose.yml
+5 -1
View File
@@ -1,7 +1,7 @@
services: services:
skin-py-analysis: skin-py-analysis:
build: /data/www/pifu build: /data/www/pifu
image: analysis-py:latest image: skin-py-gpu:1.0.0
working_dir: /app working_dir: /app
ports: ports:
- "8000:8000" - "8000:8000"
@@ -16,8 +16,12 @@ services:
MAX_PENDING_TASKS: 100 MAX_PENDING_TASKS: 100
MODEL_PATH: /app/best.pt MODEL_PATH: /app/best.pt
IMAGE_WORKERS: 4 IMAGE_WORKERS: 4
NVIDIA_VISIBLE_DEVICES: all
NVIDIA_DRIVER_CAPABILITIES: compute,utility
command: python api_server.py command: python api_server.py
restart: unless-stopped restart: unless-stopped
gpus: all
networks: networks:
- default - default
- common-network - common-network