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