Fixing torch compile benchmark (#3179)

* adding no dependency flag to torchtext * adding torchtext back --------- Co-authored-by: Ankith Gunapal <[email protected]>
pytorch · Jun 7, 2024 · 36049cb · 36049cb
1 parent a10aa45
commit 36049cb
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/ts_scripts/install_dependencies.py b/ts_scripts/install_dependencies.py
@@ -138,10 +138,7 @@ def install_python_packages(self, cuda_version, requirements_file_path, nightly)
         if nightly:
             pt_nightly = "cpu" if not cuda_version else cuda_version
             os.system(
-                f"pip3 install numpy --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/{pt_nightly}"
-            )
-            os.system(
-                f"pip3 install --pre torchtext --index-url https://download.pytorch.org/whl/nightly/cpu"
+                f"pip3 install numpy --pre torch torchvision torchaudio torchtext --index-url https://download.pytorch.org/whl/nightly/{pt_nightly}"
             )
         else:
             self.install_torch_packages(cuda_version)