init

docker-compose.yml

@@ -8,6 +8,13 @@ services:
       - "8000:8000"
     volumes:
       - .:/app
-      - ./model:/app/model   # 挂载本地的模型目录
+      - ./model:/app/model   # 挂载模型
     environment:
       - PYTHONUNBUFFERED=1
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]