From 7c8db2887e463e8b16ffbd90049f8e0dfcd4c22a Mon Sep 17 00:00:00 2001 From: Masahiro Hiramori Date: Mon, 28 Apr 2025 13:56:18 +0900 Subject: [PATCH] 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 547f51fbb493..dcf2f2c42192 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 \