(tvm) branch main updated: [CI] Upgrade PyTorch to 2.4.1 (#17338)

[tqchen]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 9f281758e8 [CI] Upgrade PyTorch to 2.4.1 (#17338)
9f281758e8 is described below

commit 9f281758e8a1a3c1c649b995367b0166da55f2c6
Author: Masahiro Hiramori <mhg00...@gmail.com>
AuthorDate: Tue Sep 17 23:07:22 2024 +0900

    [CI] Upgrade PyTorch to 2.4.1 (#17338)
    
    upgrade pytorch to 2.4.1
---
 docker/install/ubuntu_install_onnx.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker/install/ubuntu_install_onnx.sh 
b/docker/install/ubuntu_install_onnx.sh
index 2bb50c6198..6cea0075c1 100755
--- a/docker/install/ubuntu_install_onnx.sh
+++ b/docker/install/ubuntu_install_onnx.sh
@@ -36,6 +36,6 @@ pip3 install \
 pip3 install future
 
 pip3 install \
-    torch==2.0.0 \
-    torchvision==0.15.1 \
+    torch==2.4.1 \
+    torchvision==0.19.1 \
     --extra-index-url https://download.pytorch.org/whl/cpu