File tree Expand file tree Collapse file tree 3 files changed +20
-24
lines changed
.ci/docker/ci_commit_pins Expand file tree Collapse file tree 3 files changed +20
-24
lines changed Original file line number Diff line number Diff line change 1
- 53a2908a10f414a2f85caa06703a26a40e873869
1
+ 7d06d03434c7c8de8fe1a432ee454b953f84ab0d
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
15
+ from torch_pin import NIGHTLY_VERSION , SUPPORTED_CUDA_VERSIONS , TORCH_VERSION
16
16
17
17
# The pip repository that hosts nightly torch packages.
18
18
# This will be dynamically set based on CUDA availability and CUDA backend enabled/disabled.
19
19
TORCH_NIGHTLY_URL_BASE = "https://download.pytorch.org/whl/nightly"
20
20
21
- # Supported CUDA versions - modify this to add/remove supported versions
22
- # Format: tuple of (major, minor) version numbers
23
- SUPPORTED_CUDA_VERSIONS = (
24
- (12 , 6 ),
25
- (12 , 8 ),
26
- (13 , 0 ),
27
- )
28
-
29
- # Since ExecuTorch often uses main-branch features of pytorch, only the nightly
30
- # pip versions will have the required features.
31
- #
32
- # NOTE: If a newly-fetched version of the executorch repo changes the value of
33
- # NIGHTLY_VERSION, you should re-run this script to install the necessary
34
- # package versions.
35
- #
36
- # NOTE: If you're changing, make the corresponding change in .ci/docker/ci_commit_pins/pytorch.txt
37
- # by picking the hash from the same date in https://hud.pytorch.org/hud/pytorch/pytorch/nightly/
38
- #
39
- # NOTE: If you're changing, make the corresponding supported CUDA versions in
40
- # SUPPORTED_CUDA_VERSIONS above if needed.
41
-
42
21
43
22
def install_requirements (use_pytorch_nightly ):
44
23
# Skip pip install on Intel macOS if using nightly.
Original file line number Diff line number Diff line change
1
+ # Since ExecuTorch often uses main-branch features of pytorch, only the nightly
2
+ # pip versions will have the required features.
3
+ #
4
+ # NOTE: If a newly-fetched version of the executorch repo changes the value of
5
+ # NIGHTLY_VERSION, you should re-run install_executorch.sh script to install the necessary
6
+ # package versions.
7
+ #
8
+ # NOTE: If you're changing, make the corresponding change in .ci/docker/ci_commit_pins/pytorch.txt
9
+ # by picking the hash from the same date in https://hud.pytorch.org/hud/pytorch/pytorch/nightly/
10
+ #
11
+ # NOTE: If you're changing, make the corresponding supported CUDA versions in
12
+ # SUPPORTED_CUDA_VERSIONS above if needed.
1
13
TORCH_VERSION = "2.10.0"
2
- NIGHTLY_VERSION = "dev20251003"
14
+ NIGHTLY_VERSION = "dev20251008"
15
+ SUPPORTED_CUDA_VERSIONS = (
16
+ (12 , 6 ),
17
+ (12 , 8 ),
18
+ (13 , 0 ),
19
+ )
You can’t perform that action at this time.
0 commit comments