Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 7 additions & 2 deletions install_requirements.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,8 @@

from install_utils import determine_torch_url, is_intel_mac_os, python_is_compatible

from torch_pin import NIGHTLY_VERSION, TORCH_VERSION

# The pip repository that hosts nightly torch packages.
# This will be dynamically set based on CUDA availability and CUDA backend enabled/disabled.
TORCH_NIGHTLY_URL_BASE = "https://download.pytorch.org/whl/nightly"
Expand All @@ -36,7 +38,6 @@
#
# NOTE: If you're changing, make the corresponding supported CUDA versions in
# SUPPORTED_CUDA_VERSIONS above if needed.
NIGHTLY_VERSION = "dev20250915"


def install_requirements(use_pytorch_nightly):
Expand All @@ -57,7 +58,11 @@ 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.10.0.{NIGHTLY_VERSION}" if use_pytorch_nightly else "torch",
(
f"torch=={TORCH_VERSION}.{NIGHTLY_VERSION}"
if use_pytorch_nightly
else "torch"
),
]

# Install the requirements for core ExecuTorch package.
Expand Down
2 changes: 2 additions & 0 deletions torch_pin.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
TORCH_VERSION = "2.10.0"
NIGHTLY_VERSION = "dev20250915"
Loading