Function install_optional_example_requirements

install_requirements.py:110–149

(use_pytorch_nightly)

def install_optional_example_requirements(use_pytorch_nightly):
    # Determine the appropriate PyTorch URL based on CUDA delegate status
    torch_url = determine_torch_url(TORCH_URL_BASE)

    print("Installing torch domain libraries")
    DOMAIN_LIBRARIES = [
        ("torchvision==0.27.0" if use_pytorch_nightly else "torchvision"),
        ("torchaudio==2.11.0" if use_pytorch_nightly else "torchaudio"),
    ]
    # Then install domain libraries
    subprocess.run(
        [
            sys.executable,
            "-m",
            "pip",
            "install",
            "--no-cache-dir",
            *DOMAIN_LIBRARIES,
            "--extra-index-url",
            torch_url,
        ],
        check=True,
    )

    print("Installing packages in requirements-examples.txt")
    subprocess.run(
        [
            sys.executable,
            "-m",
            "pip",
            "install",
            "-r",
            "requirements-examples.txt",
            "--extra-index-url",
            torch_url,
            "--upgrade-strategy",
            "only-if-needed",
        ],
        check=True,
    )


def main(args):