testing torchtext

pytorch · Jun 6, 2024 · 255ca3f · 255ca3f
1 parent 294dcb8
commit 255ca3f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ts_scripts/install_dependencies.py b/ts_scripts/install_dependencies.py
@@ -138,11 +138,11 @@ 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 --no-deps"
+                f"pip3 install numpy --pre torch torchvision torchaudio torchtext --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 --no-deps"
+            # )
         else:
             self.install_torch_packages(cuda_version)