Remove torchaudio and torchvision from core dependencies

Author: mergennachin

install_requirements.py

@@ -72,14 +72,6 @@ def install_requirements(use_pytorch_nightly):
         )
         sys.exit(1)
 
-    # pip packages needed by exir.
-    TORCH_PACKAGE = [
-        # Setting use_pytorch_nightly to false to test the pinned PyTorch commit. Note
-        # 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",
-    ]
-
     # Install the requirements for core ExecuTorch package.
     # `--extra-index-url` tells pip to look for package
     # versions on the provided URL if they aren't available on the default URL.
@@ -91,7 +83,6 @@ def install_requirements(use_pytorch_nightly):
             "install",
             "-r",
             "requirements-dev.txt",
-            *TORCH_PACKAGE,
             "--extra-index-url",
             TORCH_URL,
         ],
@@ -130,25 +121,6 @@ def install_requirements(use_pytorch_nightly):
 
 
 def install_optional_example_requirements(use_pytorch_nightly):
-    print("Installing torch domain libraries")
-    DOMAIN_LIBRARIES = [
-        ("torchvision==0.23.0" if use_pytorch_nightly else "torchvision"),
-        "torchaudio==2.8.0" if use_pytorch_nightly else "torchaudio",
-    ]
-    # Then install domain libraries
-    subprocess.run(
-        [
-            sys.executable,
-            "-m",
-            "pip",
-            "install",
-            *DOMAIN_LIBRARIES,
-            "--extra-index-url",
-            TORCH_URL,
-        ],
-        check=True,
-    )
-
     print("Installing packages in requirements-examples.txt")
     subprocess.run(
         [
@@ -158,10 +130,6 @@ def install_optional_example_requirements(use_pytorch_nightly):
             "install",
             "-r",
             "requirements-examples.txt",
-            "--extra-index-url",
-            TORCH_URL,
-            "--upgrade-strategy",
-            "only-if-needed",
         ],
         check=True,
     )

pyproject.toml

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

requirements-examples.txt

@@ -5,3 +5,5 @@ timm == 1.0.7
 torchsr == 1.0.4
 torchtune >= 0.6.1
 transformers ==4.47.1
+torchaudio>=2.8.0,<2.9.0
+torchvision>=0.23.0,<0.24.0