From ef2a125d7ec35c9f26a58d7a917a595114400147 Mon Sep 17 00:00:00 2001 From: Jack-Khuu Date: Wed, 2 Oct 2024 13:05:13 -0700 Subject: [PATCH] Update install_requirements.py: Bumping PT pin to dev20240925 (#5824) Summary: Bumping it to Pytorch 2.6 to pick up changes for AOTI packaging and distributed folk Recreated Version of https://github.com/pytorch/executorch/pull/5549 Differential Revision: D63776738 Pulled By: Jack-Khuu --- .ci/docker/ci_commit_pins/pytorch.txt | 2 +- install_requirements.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.ci/docker/ci_commit_pins/pytorch.txt b/.ci/docker/ci_commit_pins/pytorch.txt index 0e9181ac55a..3c387a6a324 100644 --- a/.ci/docker/ci_commit_pins/pytorch.txt +++ b/.ci/docker/ci_commit_pins/pytorch.txt @@ -1 +1 @@ -aec9b2ab77389967ef39bb9c10662fd0fe3e185a +5ba404f68775bb06a1125a100687f86b6d6de6a8 diff --git a/install_requirements.py b/install_requirements.py index f169c52257c..6563b395e80 100644 --- a/install_requirements.py +++ b/install_requirements.py @@ -94,14 +94,14 @@ def python_is_compatible(): # NOTE: If a newly-fetched version of the executorch repo changes the value of # NIGHTLY_VERSION, you should re-run this script to install the necessary # package versions. -NIGHTLY_VERSION = "dev20240912" +NIGHTLY_VERSION = "dev20240927" # The pip repository that hosts nightly torch packages. TORCH_NIGHTLY_URL = "https://download.pytorch.org/whl/nightly/cpu" # pip packages needed by exir. EXIR_REQUIREMENTS = [ - f"torch==2.5.0.{NIGHTLY_VERSION}", + f"torch==2.6.0.{NIGHTLY_VERSION}", f"torchvision==0.20.0.{NIGHTLY_VERSION}", # For testing. "typing-extensions", ]