优化包的保存方式

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 \ 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 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 # EXPOSE 8081

View File

@@ -8,6 +8,7 @@ services:
volumes: volumes:
- ./log:/app/log - ./log:/app/log
- ./model:/app/model - ./model:/app/model
- ./packages:/app/packages
restart: always restart: always
command: [ "photo_review.py", "--clean", "True" ] command: [ "photo_review.py", "--clean", "True" ]
deploy: deploy:

1
model/README.md Normal file
View File

@@ -0,0 +1 @@
存放深度学习模型的目录

1
packages/README.md Normal file
View File

@@ -0,0 +1 @@
存放依赖包的目录