为照片识别编写docker

2024-07-18 11:19:21 +08:00
parent 31256a6271
commit 74e9989f34
3 changed files with 114 additions and 9 deletions
--- a/4
+++ b/4
@@ -1,3 +1,4 @@
+ARG VERSION
 # 使用官方的paddle镜像作为基础
 FROM registry.baidubce.com/paddlepaddle/paddle:2.6.1-gpu-cuda12.0-cudnn8.9-trt8.6

@@ -5,7 +6,8 @@ FROM registry.baidubce.com/paddlepaddle/paddle:2.6.1-gpu-cuda12.0-cudnn8.9-trt8.
 WORKDIR /app

 # 设置环境变量
-ENV PYTHONUNBUFFERED=1 \
+ENV VERSION=${VERSION} \
+    PYTHONUNBUFFERED=1 \
    # 设置pip镜像地址，加快安装速度
    PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple