File tree Expand file tree Collapse file tree 1 file changed +18
-0
lines changed
Expand file tree Collapse file tree 1 file changed +18
-0
lines changed Original file line number Diff line number Diff line change @@ -143,6 +143,24 @@ def install_optional_example_requirements(use_pytorch_nightly):
143143 check = True ,
144144 )
145145
146+ print ("Installing torch domain libraries" )
147+ DOMAIN_LIBRARIES = [
148+ ("torchvision==0.23.0" if use_pytorch_nightly else "torchvision" ),
149+ "torchaudio==2.8.0" if use_pytorch_nightly else "torchaudio" ,
150+ ]
151+ # Then install domain libraries
152+ subprocess .run (
153+ [
154+ sys .executable ,
155+ "-m" ,
156+ "pip" ,
157+ "install" ,
158+ * DOMAIN_LIBRARIES ,
159+ "--extra-index-url" ,
160+ TORCH_URL ,
161+ ],
162+ check = True ,
163+ )
146164
147165# Prebuilt binaries for Intel-based macOS are no longer available on PyPI; users must compile from source.
148166# PyTorch stopped building macOS x86_64 binaries since version 2.3.0 (January 2024).
You can’t perform that action at this time.
0 commit comments