remove torchtext in the installation, which causes endless version fe…

…tching
GoogleCloudPlatform · Feb 9, 2024 · 4cce2ee · 4cce2ee
1 parent 693ee3f
commit 4cce2ee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dags/pytorch_xla/configs/pytorchxla_torchbench_config.py b/dags/pytorch_xla/configs/pytorchxla_torchbench_config.py
@@ -162,8 +162,8 @@ def get_nvidia_driver_install_cmd(driver_version: str) -> str:
 
   docker_cmds_ls = (
       "apt-get update && apt-get install -y libgl1",
-      "pip install --user numpy pandas",
-      "pip install --user --pre torch torchvision torchaudio torchtext -i https://download.pytorch.org/whl/nightly/cu121",
+      "pip3 install --user numpy pandas",
+      "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121"
       "cd /tmp/ && git clone https://github.com/pytorch/benchmark.git",
       f" cd benchmark && {model_install_cmds()}",
       "cd /tmp/ && git clone https://github.com/pytorch/pytorch.git",