Skip to content

Commit 268e615

Browse files
committed
Fix again
1 parent 679e1ad commit 268e615

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

install_requirements.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def install_requirements(use_pytorch_release):
7777
# Setting use_pytorch_nightly to false to test the pinned PyTorch commit. Note
7878
# that we don't need to set any version number there because they have already
7979
# been installed on CI before this step, so pip won't reinstall them
80-
"torch==2.8.0" if use_pytorch_release else "torch",
80+
"torch==2.8.0+cpu" if use_pytorch_release else "torch",
8181
]
8282

8383
# Install the requirements for core ExecuTorch package.
@@ -145,8 +145,8 @@ def install_optional_example_requirements(use_pytorch_release):
145145

146146
print("Installing torch domain libraries")
147147
DOMAIN_LIBRARIES = [
148-
"torchvision==0.23.0" if use_pytorch_release else "torchvision",
149-
"torchaudio==2.8.0" if use_pytorch_release else "torchaudio",
148+
"torchvision==0.23.0+cpu" if use_pytorch_release else "torchvision",
149+
"torchaudio==2.8.0+cpu" if use_pytorch_release else "torchaudio",
150150
]
151151
# Then install domain libraries
152152
subprocess.run(

0 commit comments

Comments
 (0)