From e2ba298e617df5f1700e0ca243b6921986545700 Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Tue, 27 Aug 2024 14:44:18 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E9=95=9C=E5=83=8F=E6=9E=84?=
 =?UTF-8?q?=E5=BB=BA=EF=BC=8C=E5=88=A0=E9=99=A4=E5=A4=9A=E4=BD=99=E7=9A=84?=
 =?UTF-8?q?torch?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Dockerfile | 12 +-----------
 1 file changed, 1 insertion(+), 11 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index c84dcf0..39a4e3c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -17,17 +17,7 @@ COPY packages /app/packages
 RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \
     && 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/ \
-    && pip install packages/torch-2.4.0+cu118-cp310-cp310-linux_x86_64.whl \
-    && pip install packages/torchvision-0.19.0+cu118-cp310-cp310-linux_x86_64.whl \
-    && pip install packages/torchaudio-2.4.0+cu118-cp310-cp310-linux_x86_64.whl
-
-# 编译文档检测的部分内容
-COPY object_detection /app/object_detection
-RUN cd /app/object_detection/core/models/py_utils/_cpools \
-    && python setup.py install --user \
-    && cd /app/object_detection/core/external \
-    && make
+    && pip install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
 
 # 将当前目录内容复制到容器的/app内
 COPY . /app