File tree Expand file tree Collapse file tree 1 file changed +9
-6
lines changed Expand file tree Collapse file tree 1 file changed +9
-6
lines changed Original file line number Diff line number Diff line change 5
5
6
6
def install_torch_nightly_deps ():
7
7
"""Install torch related dependencies from pinned nightly"""
8
- NIGHTLY_VERSION = "dev20250523"
8
+ EXECUTORCH_NIGHTLY_VERSION = "dev20250620"
9
+ TORCHAO_NIGHTLY_VERSION = "dev20250620"
10
+ # Torch nightly is aligned with pinned nightly in https://github.com/pytorch/executorch/blob/main/install_requirements.py#L74
11
+ TORCH_NIGHTLY_VERSION = "dev20250601"
9
12
subprocess .check_call (
10
13
[
11
14
sys .executable ,
12
15
"-m" ,
13
16
"pip" ,
14
17
"install" ,
15
- f"executorch==0.7.0.{ NIGHTLY_VERSION } " ,
16
- f"torch==2.8.0.{ NIGHTLY_VERSION } " ,
17
- f"torchvision==0.22 .0.{ NIGHTLY_VERSION } " ,
18
- f"torchaudio==2.6 .0.{ NIGHTLY_VERSION } " ,
19
- "torchao==0.12.0.dev20250528 " ,
18
+ f"executorch==0.7.0.{ EXECUTORCH_NIGHTLY_VERSION } " ,
19
+ f"torch==2.8.0.{ TORCH_NIGHTLY_VERSION } " ,
20
+ f"torchvision==0.23 .0.{ TORCH_NIGHTLY_VERSION } " ,
21
+ f"torchaudio==2.8 .0.{ TORCH_NIGHTLY_VERSION } " ,
22
+ f "torchao==0.12.0.{ TORCHAO_NIGHTLY_VERSION } " ,
20
23
"--extra-index-url" ,
21
24
"https://download.pytorch.org/whl/nightly/cpu" ,
22
25
]
You can’t perform that action at this time.
0 commit comments