23 lines
465 B
Docker
23 lines
465 B
Docker
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
|