1

2026-08-17 10:03:37 +08:00
parent 350223c5ad
commit f0ccc5b2aa
18 changed files with 12222 additions and 0 deletions
@@ -0,0 +1,22 @@
+FROM skin-py-cpu-base:v1.0.2
+
+WORKDIR /app
+
+ENV PYTHONUNBUFFERED=1
+ENV PIP_NO_CACHE_DIR=1
+
+COPY wanzheng.py .
+COPY api_server.py .
+COPY config ./config
+COPY best.pt .
+COPY shape_predictor_68_face_landmarks.dat .
+
+EXPOSE 8071
+
+CMD ["python", "api_server.py"]
+
+# 服务器执行 nvidia-smi 查看CUDA版本 比如 CUDA 12.1 就是/whl/cu121
+# RUN pip install \
+#     torch==2.4.1 \
+#     torchvision==0.19.1 \
+#     --index-url https://download.pytorch.org/whl/cu121