From 9bd7038d5f87415e885bae84ee66c6c557d6adda Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Thu, 18 Jul 2024 15:24:20 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4docker=E4=B8=AD=E4=BE=9D?=
 =?UTF-8?q?=E8=B5=96=E7=9A=84=E5=AE=89=E8=A3=85?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Dockerfile         | 4 +++-
 docker-compose.yml | 2 +-
 requirements.txt   | 2 ++
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 6990ed0..916c5d3 100644
--- a/Dockerfile
+++ b/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
diff --git a/docker-compose.yml b/docker-compose.yml
index 6146dbd..4249960 100644
--- a/docker-compose.yml
+++ b/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
diff --git a/requirements.txt b/requirements.txt
index 0644837..e807d3f 100644
--- a/requirements.txt
+++ b/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