File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff 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 (
You can’t perform that action at this time.
0 commit comments