This is an automated email from the ASF dual-hosted git repository.
yongwww 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 03eadc6551 [CI] Install PyTorch 2.7 compatible with CUDA 11.8 (#17905)
03eadc6551 is described below
commit 03eadc655181b1a33e83faf37e1c8d289a9a94f7
Author: Masahiro Hiramori <[email protected]>
AuthorDate: Mon Apr 28 15:41:32 2025 +0900
[CI] Install PyTorch 2.7 compatible with CUDA 11.8 (#17905)
use torch cuda 11.8
---
docker/install/ubuntu_install_onnx.sh | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/docker/install/ubuntu_install_onnx.sh
b/docker/install/ubuntu_install_onnx.sh
index 547f51fbb4..dcf2f2c421 100755
--- a/docker/install/ubuntu_install_onnx.sh
+++ b/docker/install/ubuntu_install_onnx.sh
@@ -45,7 +45,8 @@ if [ "$PYTHON_VERSION" == "3.9" ]; then
if [ "$DEVICE" == "cuda" ]; then
pip3 install \
torch==2.7.0 \
- torchvision==0.22.0
+ torchvision==0.22.0 \
+ --index-url https://download.pytorch.org/whl/cu118
else
pip3 install \
torch==2.7.0 \
@@ -61,7 +62,8 @@ elif [ "$PYTHON_VERSION" == "3.11" ]; then
if [ "$DEVICE" == "cuda" ]; then
pip3 install \
torch==2.7.0 \
- torchvision==0.22.0
+ torchvision==0.22.0 \
+ --index-url https://download.pytorch.org/whl/cu118
else
pip3 install \
torch==2.7.0 \
