From 4a7c3d85095bc4148098db45efc90389db40a42a Mon Sep 17 00:00:00 2001 From: Scott Roy <161522778+metascroy@users.noreply.github.com> Date: Fri, 28 Mar 2025 11:52:32 -0700 Subject: [PATCH 1/2] init --- install_requirements.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/install_requirements.py b/install_requirements.py index 4b3de68b8db..64b7b902299 100644 --- a/install_requirements.py +++ b/install_requirements.py @@ -59,8 +59,7 @@ def python_is_compatible(): # The pip repository that hosts nightly torch packages. -TORCH_NIGHTLY_URL = "https://download.pytorch.org/whl/nightly/cpu" - +TORCH_URL = "https://download.pytorch.org/whl/test/cpu" # Since ExecuTorch often uses main-branch features of pytorch, only the nightly # pip versions will have the required features. @@ -80,16 +79,14 @@ def install_requirements(use_pytorch_nightly): # 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 - f"torch==2.7.0.{NIGHTLY_VERSION}" if use_pytorch_nightly else "torch", + "torch==2.7.0" if use_pytorch_nightly else "torch", ( - f"torchvision==0.22.0.{NIGHTLY_VERSION}" - if use_pytorch_nightly - else "torchvision" + "torchvision==0.23.0" if use_pytorch_nightly else "torchvision" ), # For testing. ] EXAMPLES_REQUIREMENTS = [ - f"torchaudio==2.6.0.{NIGHTLY_VERSION}" if use_pytorch_nightly else "torchaudio", + "torchaudio==2.7.0" if use_pytorch_nightly else "torchaudio", ] # Assemble the list of requirements to actually install. @@ -110,7 +107,7 @@ def install_requirements(use_pytorch_nightly): "requirements-dev.txt", *REQUIREMENTS_TO_INSTALL, "--extra-index-url", - TORCH_NIGHTLY_URL, + TORCH_URL, ], check=True, ) From 730c343bb4bb04d4cc9af30a709c91e7d7660f07 Mon Sep 17 00:00:00 2001 From: Scott Roy <161522778+metascroy@users.noreply.github.com> Date: Fri, 28 Mar 2025 12:18:35 -0700 Subject: [PATCH 2/2] up --- install_executorch.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/install_executorch.py b/install_executorch.py index 85703903ffc..491ba19dec1 100644 --- a/install_executorch.py +++ b/install_executorch.py @@ -17,11 +17,7 @@ from contextlib import contextmanager from typing import List, Tuple -from install_requirements import ( - install_requirements, - python_is_compatible, - TORCH_NIGHTLY_URL, -) +from install_requirements import install_requirements, python_is_compatible, TORCH_URL # Set up logging logging.basicConfig( @@ -260,7 +256,7 @@ def main(args): "--no-build-isolation", "-v", "--extra-index-url", - TORCH_NIGHTLY_URL, + TORCH_URL, ], check=True, )