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