Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

调整docker中依赖的安装

Browse Source
This commit is contained in:
liuyebo
2024-07-18 15:24:20 +08:00
parent cb60802d24
commit 9bd7038d5f
3 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile
View File

@@ -15,7 +15,9 @@ ENV VERSION=${VERSION} \
COPY . /app COPY . /app
# 安装任何需要的包 # 安装任何需要的包
RUN pip install --no-cache-dir -r requirements.txt RUN 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/
# 暴露端口 # 暴露端口
# EXPOSE 8081 # EXPOSE 8081

2
docker-compose.yml
View File

@@ -5,7 +5,7 @@ services:
build: build:
context: . context: .
args: args:
VERSION: "0.0.2" VERSION: "0.0.3"
volumes: volumes:
- ./log:/app/log - ./log:/app/log
restart: always restart: always

2
requirements.txt
View File

@@ -1,5 +1,7 @@
numpy==1.26.2 numpy==1.26.2
onnxconverter-common==1.14.0
opencv-python==4.6.0.66 opencv-python==4.6.0.66
paddle2onnx==1.2.3
paddleclas==2.5.2 paddleclas==2.5.2
paddlenlp==2.6.1 paddlenlp==2.6.1
paddleocr==2.7.3 paddleocr==2.7.3