init

Dockerfile

@@ -1,4 +1,4 @@
-FROM python3.11:slim
+FROM python:3.11-slim
 
 WORKDIR /app
 

docker-compose.yml

@@ -14,13 +14,13 @@ services:
       - PYTHONUNBUFFERED=1
       - PIP_DISABLE_PIP_VERSION_CHECK=on
     # 如果需要GPU支持，取消下面的注释
-    # deploy:
-    #   resources:
-    #     reservations:
-    #       devices:
-    #         - driver: nvidia
-    #           count: 1
-    #           capabilities: [gpu]
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
     # 如果需要挂载模型数据卷，取消下面的注释
     # volumes:
     #   - ./models:/app/models