From 2a7bf5626bed0b215f0f9648d7ddbdc66614caf6 Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Thu, 22 Aug 2024 11:23:48 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4torch=E7=89=88=E6=9C=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index f612843..d429fbf 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -18,7 +18,7 @@ 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+cu121-cp310-cp310-linux_x86_64.whl \
+    && 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