This is an automated email from the ASF dual-hosted git repository. mshr pushed a commit to branch pytorch-2.7 in repository https://gitbox.apache.org/repos/asf/tvm.git
commit b33492a4381fb8787b5226c69642ed4aeeac29da Author: Masahiro Hiramori <[email protected]> AuthorDate: Thu Apr 24 19:06:47 2025 +0900 upgrade pytorch and torchvision --- docker/install/ubuntu_install_onnx.sh | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/docker/install/ubuntu_install_onnx.sh b/docker/install/ubuntu_install_onnx.sh index dc41c39d7c..547f51fbb4 100755 --- a/docker/install/ubuntu_install_onnx.sh +++ b/docker/install/ubuntu_install_onnx.sh @@ -44,12 +44,12 @@ if [ "$PYTHON_VERSION" == "3.9" ]; then if [ "$DEVICE" == "cuda" ]; then pip3 install \ - torch==2.6.0 \ - torchvision==0.21.0 + torch==2.7.0 \ + torchvision==0.22.0 else pip3 install \ - torch==2.6.0 \ - torchvision==0.21.0 \ + torch==2.7.0 \ + torchvision==0.22.0 \ --extra-index-url https://download.pytorch.org/whl/cpu fi elif [ "$PYTHON_VERSION" == "3.11" ]; then @@ -60,12 +60,12 @@ elif [ "$PYTHON_VERSION" == "3.11" ]; then if [ "$DEVICE" == "cuda" ]; then pip3 install \ - torch==2.6.0 \ - torchvision==0.21.0 + torch==2.7.0 \ + torchvision==0.22.0 else pip3 install \ - torch==2.6.0 \ - torchvision==0.21.0 \ + torch==2.7.0 \ + torchvision==0.22.0 \ --extra-index-url https://download.pytorch.org/whl/cpu fi else
