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 \