fix: use gpu

2026-09-15 10:01:41 +08:00
parent a9d7aac95c
commit 9563d7bb98
2 changed files with 6 additions and 2 deletions
@@ -1,4 +1,4 @@
-FROM skin-py-cpu-base:v1.0.2
+FROM skin-py-gpu:1.0.0

 WORKDIR /app

@@ -1,7 +1,7 @@
 services:
  skin-py-analysis:
    build: /data/www/pifu
-    image: analysis-py:latest
+    image: skin-py-gpu:1.0.0
    working_dir: /app
    ports:
      - "8000:8000"
@@ -16,8 +16,12 @@ services:
      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