torch==2.9.0

Author: mshr-h

docker/install/ubuntu_install_onnx.sh

@@ -44,12 +44,12 @@ if [ "$PYTHON_VERSION" == "3.10" ]; then
 
     if [ "$DEVICE" == "cuda" ]; then
         pip3 install \
-            torch==2.9.1 \
+            torch==2.9.0 \
             torchvision==0.24.0 \
             --index-url https://download.pytorch.org/whl/cu118
     else
         pip3 install \
-            torch==2.9.1 \
+            torch==2.9.0 \
             torchvision==0.24.0 \
             --extra-index-url https://download.pytorch.org/whl/cpu
     fi
@@ -61,12 +61,12 @@ elif [ "$PYTHON_VERSION" == "3.11" ]; then
 
     if [ "$DEVICE" == "cuda" ]; then
         pip3 install \
-            torch==2.9.1 \
+            torch==2.9.0 \
             torchvision==0.24.0 \
             --index-url https://download.pytorch.org/whl/cu118
     else
         pip3 install \
-            torch==2.9.1 \
+            torch==2.9.0 \
             torchvision==0.24.0 \
             --extra-index-url https://download.pytorch.org/whl/cpu
     fi