Skip to content

Commit 6171d02

Browse files
committed
Try to fix install wheel job failure
1 parent e91806e commit 6171d02

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

install_requirements.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff 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).

0 commit comments

Comments
 (0)