优化包的保存方式

This commit is contained in:
2024-08-09 09:19:02 +08:00
parent dcf9c80256
commit 42c4aba1f7
4 changed files with 7 additions and 2 deletions

View File

@@ -16,9 +16,11 @@ COPY . /app
# 安装任何需要的包
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \
&& pip install --no-cache-dir --find-links=./packages -r requirements.txt \
&& 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 install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
&& 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