[Build] Lock torch, onnxscript and onnx-ir versions to latest (microsoft#26315)

To fix build pipeline error `ModuleNotFoundError: No module named
'onnxscript._framework_apis.torch_2_9'` after recent torch 2.9 release.

This locks torch version to 2.8, and also updates onnxscript and onnx-ir
to latest versions.

I locked torchvision version since it is usually installed with torch
together. If torch and torchvision are not compatible, there might be
errors in transformers script.

Author: tianleiwu

tools/ci_build/build.py

@@ -1751,7 +1751,7 @@ def run_onnxruntime_tests(args, source_dir, ctest_path, build_dir, configs):
                 # Install cpu only version of torch when cuda is not enabled in Linux.
                 extra = [] if args.use_cuda and is_linux() else ["--index-url", "https://download.pytorch.org/whl/cpu"]
                 run_subprocess(
-                    [sys.executable, "-m", "pip", "install", "torch", *extra],
+                    [sys.executable, "-m", "pip", "install", "torch==2.8.0", "torchvision==0.23.0", *extra],
                     cwd=cwd,
                     dll_path=dll_path,
                     python_path=python_path,

tools/ci_build/github/linux/docker/scripts/requirements.txt

@@ -10,5 +10,5 @@ sympy==1.14
 flatbuffers
 protobuf==4.25.1
 packaging
-onnxscript==0.3.2
-onnx-ir
+onnxscript==0.5.3
+onnx-ir==0.1.10

tools/ci_build/github/linux/python/requirements.txt

@@ -8,7 +8,7 @@ protobuf==4.25.1
 sympy==1.14
 flatbuffers
 psutil
-onnxscript==0.3.2
-onnx-ir
+onnxscript==0.5.3
+onnx-ir==0.1.10
 jinja2
 markupsafe

tools/ci_build/github/windows/python/requirements.txt

@@ -8,8 +8,8 @@ protobuf==4.25.1
 sympy==1.14
 flatbuffers
 psutil
-onnxscript==0.3.2
-onnx-ir
+onnxscript==0.5.3
+onnx-ir==0.1.10
 jinja2
 markupsafe
 semver

tools/ci_build/requirements/transformers-test/requirements.txt

@@ -3,12 +3,13 @@ packaging
 # protobuf and numpy is same as tools/ci_build/github/linux/docker/scripts/manylinux/requirements.txt
 protobuf==4.25.1
 numpy==2.2.6
-torch>=2.6.0
+torch==2.8.0
+torchvision==0.23.0
 coloredlogs==15.0
 transformers==4.52.1
 parameterized>=0.8.1
 sentencepiece
 psutil
 einops
-onnxscript==0.3.2
-onnx-ir
+onnxscript==0.5.3
+onnx-ir==0.1.10