Use Torch-stable 2.3.1

Xilinx · Jul 24, 2024 · b9fae35 · b9fae35
1 parent 6bb6554
commit b9fae35
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/build_tools/ci/install_python_deps.sh b/build_tools/ci/install_python_deps.sh
@@ -19,7 +19,7 @@ case $torch_version in
     ;;
   stable)
     echo "::group::installing stable torch"
-    python3 -m pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu
+    python3 -m pip install --no-cache-dir torch==2.3.1 torchvision --index-url https://download.pytorch.org/whl/cpu
     python3 -m pip install --no-cache-dir -r $repo_root/build-requirements.txt
     echo "::endgroup::"
     ;;