调整docker中依赖的安装

Dockerfile

@@ -15,7 +15,9 @@ ENV VERSION=${VERSION} \
 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

docker-compose.yml

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

requirements.txt

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