Skip to content

Commit b9c4ebd

Browse files
committed
Try fixing torch and domain library conflicts
1 parent fb61bb5 commit b9c4ebd

File tree

2 files changed

+13
-11
lines changed

2 files changed

+13
-11
lines changed

install_requirements.py

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -130,32 +130,34 @@ def install_requirements(use_pytorch_nightly):
130130

131131

132132
def install_optional_example_requirements(use_pytorch_nightly):
133-
print("Installing packages in requirements-examples.txt")
133+
print("Installing torch domain libraries")
134+
DOMAIN_LIBRARIES = [
135+
("torchvision==0.23.0" if use_pytorch_nightly else "torchvision"),
136+
"torchaudio==2.8.0" if use_pytorch_nightly else "torchaudio",
137+
]
138+
# Then install domain libraries
134139
subprocess.run(
135140
[
136141
sys.executable,
137142
"-m",
138143
"pip",
139144
"install",
140-
"-r",
141-
"requirements-examples.txt",
145+
*DOMAIN_LIBRARIES,
146+
"--extra-index-url",
147+
TORCH_URL,
142148
],
143149
check=True,
144150
)
145151

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+
print("Installing packages in requirements-examples.txt")
152153
subprocess.run(
153154
[
154155
sys.executable,
155156
"-m",
156157
"pip",
157158
"install",
158-
*DOMAIN_LIBRARIES,
159+
"-r",
160+
"requirements-examples.txt",
159161
"--extra-index-url",
160162
TORCH_URL,
161163
],

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ dependencies=[
6969
"sympy",
7070
"torch>=2.8.0,<2.9.0",
7171
"torchaudio>=2.8.0,<2.9.0",
72-
"torchvisionn>=0.23.0,<0.24.0",
72+
"torchvision>=0.23.0,<0.24.0",
7373
"torchao==0.12.0",
7474
"tabulate",
7575
# See also third-party/TARGETS for buck's typing-extensions version.

0 commit comments

Comments
 (0)