Skip to content
This repository was archived by the owner on Aug 7, 2025. It is now read-only.

Commit 36049cb

Browse files
udaij12agunapal
andauthored
Fixing torch compile benchmark (#3179)
* adding no dependency flag to torchtext * adding torchtext back --------- Co-authored-by: Ankith Gunapal <agunapal@ischool.Berkeley.edu>
1 parent a10aa45 commit 36049cb

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

ts_scripts/install_dependencies.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -138,10 +138,7 @@ def install_python_packages(self, cuda_version, requirements_file_path, nightly)
138138
if nightly:
139139
pt_nightly = "cpu" if not cuda_version else cuda_version
140140
os.system(
141-
f"pip3 install numpy --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/{pt_nightly}"
142-
)
143-
os.system(
144-
f"pip3 install --pre torchtext --index-url https://download.pytorch.org/whl/nightly/cpu"
141+
f"pip3 install numpy --pre torch torchvision torchaudio torchtext --index-url https://download.pytorch.org/whl/nightly/{pt_nightly}"
145142
)
146143
else:
147144
self.install_torch_packages(cuda_version)

0 commit comments

Comments
 (0)