File tree Expand file tree Collapse file tree 2 files changed +9
-2
lines changed Expand file tree Collapse file tree 2 files changed +9
-2
lines changed Original file line number Diff line number Diff line change 1212
1313from install_utils import determine_torch_url , is_intel_mac_os , python_is_compatible
1414
15+ from torch_pin import NIGHTLY_VERSION , TORCH_VERSION
16+
1517# The pip repository that hosts nightly torch packages.
1618# This will be dynamically set based on CUDA availability and CUDA backend enabled/disabled.
1719TORCH_NIGHTLY_URL_BASE = "https://download.pytorch.org/whl/nightly"
3638#
3739# NOTE: If you're changing, make the corresponding supported CUDA versions in
3840# SUPPORTED_CUDA_VERSIONS above if needed.
39- NIGHTLY_VERSION = "dev20250915"
4041
4142
4243def install_requirements (use_pytorch_nightly ):
@@ -57,7 +58,11 @@ def install_requirements(use_pytorch_nightly):
5758 # Setting use_pytorch_nightly to false to test the pinned PyTorch commit. Note
5859 # that we don't need to set any version number there because they have already
5960 # been installed on CI before this step, so pip won't reinstall them
60- f"torch==2.10.0.{ NIGHTLY_VERSION } " if use_pytorch_nightly else "torch" ,
61+ (
62+ f"torch=={ TORCH_VERSION } .{ NIGHTLY_VERSION } "
63+ if use_pytorch_nightly
64+ else "torch"
65+ ),
6166 ]
6267
6368 # Install the requirements for core ExecuTorch package.
Original file line number Diff line number Diff line change 1+ TORCH_VERSION = "2.10.0"
2+ NIGHTLY_VERSION = "dev20250915"
You can’t perform that action at this time.
0 commit comments