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 12
12
13
13
from install_utils import determine_torch_url , is_intel_mac_os , python_is_compatible
14
14
15
+ from torch_pin import NIGHTLY_VERSION , TORCH_VERSION
16
+
15
17
# The pip repository that hosts nightly torch packages.
16
18
# This will be dynamically set based on CUDA availability and CUDA backend enabled/disabled.
17
19
TORCH_NIGHTLY_URL_BASE = "https://download.pytorch.org/whl/nightly"
36
38
#
37
39
# NOTE: If you're changing, make the corresponding supported CUDA versions in
38
40
# SUPPORTED_CUDA_VERSIONS above if needed.
39
- NIGHTLY_VERSION = "dev20250915"
40
41
41
42
42
43
def install_requirements (use_pytorch_nightly ):
@@ -57,7 +58,11 @@ def install_requirements(use_pytorch_nightly):
57
58
# Setting use_pytorch_nightly to false to test the pinned PyTorch commit. Note
58
59
# that we don't need to set any version number there because they have already
59
60
# 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
+ ),
61
66
]
62
67
63
68
# 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