From 42c4aba1f7c1d9ed14302eb35cd4a2d925673713 Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Fri, 9 Aug 2024 09:19:02 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E5=8C=85=E7=9A=84=E4=BF=9D?=
 =?UTF-8?q?=E5=AD=98=E6=96=B9=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Dockerfile         | 6 ++++--
 docker-compose.yml | 1 +
 model/README.md    | 1 +
 packages/README.md | 1 +
 4 files changed, 7 insertions(+), 2 deletions(-)
 create mode 100644 model/README.md
 create mode 100644 packages/README.md

diff --git a/Dockerfile b/Dockerfile
index 35c6c3d..ec03b40 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -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
diff --git a/docker-compose.yml b/docker-compose.yml
index e89c7cb..6aed6b5 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -8,6 +8,7 @@ services:
     volumes:
       - ./log:/app/log
       - ./model:/app/model
+      - ./packages:/app/packages
     restart: always
     command: [ "photo_review.py", "--clean", "True" ]
     deploy:
diff --git a/model/README.md b/model/README.md
new file mode 100644
index 0000000..2de0de7
--- /dev/null
+++ b/model/README.md
@@ -0,0 +1 @@
+存放深度学习模型的目录
\ No newline at end of file
diff --git a/packages/README.md b/packages/README.md
new file mode 100644
index 0000000..c1be01b
--- /dev/null
+++ b/packages/README.md
@@ -0,0 +1 @@
+存放依赖包的目录
\ No newline at end of file