Follow release/0.6

Author: larryliu0820

install_requirements.py

@@ -78,6 +78,10 @@ def install_requirements(use_pytorch_nightly):
         # that we don't need to set any version number there because they have already
         # been installed on CI before this step, so pip won't reinstall them
         "torch==2.8.0" if use_pytorch_nightly else "torch",
+        (
+            "torchvision==0.23.0" if use_pytorch_nightly else "torchvision"
+        ),  # For testing.
+        "torchaudio==2.8.0" if use_pytorch_nightly else "torchaudio",
     ]
 
     # Install the requirements for core ExecuTorch package.

pyproject.toml

@@ -67,9 +67,9 @@ dependencies=[
   "pyyaml",
   "ruamel.yaml",
   "sympy",
-  "torch>=2.8.0,<2.9.0",
-  "torchaudio>=2.8.0,<2.9.0",
-  "torchvision>=0.23.0,<0.24.0",
+  "torch==2.8.0",
+  "torchaudio==2.8.0",
+  "torchvision==0.23.0",
   "torchao==0.12.0",
   "tabulate",
   # See also third-party/TARGETS for buck's typing-extensions version.