diff --git a/Dockerfile b/Dockerfile index 740e4f0..684bdca 100644 --- a/Dockerfile +++ b/Dockerfile @@ -4,8 +4,6 @@ FROM registry.baidubce.com/paddlepaddle/paddle:2.6.1-gpu-cuda12.0-cudnn8.9-trt8. # 设置工作目录 WORKDIR /app -VOLUME /app/packages - # 设置环境变量 ENV PYTHONUNBUFFERED=1 \ # 设置时区 @@ -13,17 +11,16 @@ ENV PYTHONUNBUFFERED=1 \ # 设置pip镜像地址,加快安装速度 PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple +# 安装依赖 +COPY requirements.txt /app/requirements.txt +RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \ + && pip install --no-cache-dir -r requirements.txt \ + && pip uninstall -y onnxruntime onnxruntime-gpu \ + && pip install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ \ + # 将当前目录内容复制到容器的/app内 COPY . /app -# 安装任何需要的包 -RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \ - && pip download -d ./packages -r requirements.txt \ - && pip install --no-cache-dir --no-index --find-links=./packages -r requirements.txt \ - && pip uninstall -y onnxruntime onnxruntime-gpu \ - && pip download -d ./packages onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ \ - && pip install --no-cache-dir --no-index --find-links=./packages onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ - # 暴露端口 # EXPOSE 8081