Try 6

Author: larryliu0820

.ci/docker/common/install_pytorch.sh

@@ -32,9 +32,9 @@ install_pytorch_and_domains() {
   pip_install "$(echo dist/*.whl)"
 
   # Grab the pinned audio and vision commits from PyTorch
-  TORCHAUDIO_VERSION=$(cat .github/ci_commit_pins/audio.txt)
+  TORCHAUDIO_VERSION=release/2.8
   export TORCHAUDIO_VERSION
-  TORCHVISION_VERSION=$(cat .github/ci_commit_pins/vision.txt)
+  TORCHVISION_VERSION=release/0.23
   export TORCHVISION_VERSION
 
   install_domains

install_requirements.py

@@ -78,10 +78,6 @@ 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.
@@ -147,25 +143,6 @@ def install_optional_example_requirements(use_pytorch_nightly):
         check=True,
     )
 
-    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,
-    )
-
 
 # Prebuilt binaries for Intel-based macOS are no longer available on PyPI; users must compile from source.
 # PyTorch stopped building macOS x86_64 binaries since version 2.3.0 (January 2024).