diff --git a/ts_scripts/install_dependencies.py b/ts_scripts/install_dependencies.py index f32d456f46..67f4fad869 100644 --- a/ts_scripts/install_dependencies.py +++ b/ts_scripts/install_dependencies.py @@ -133,7 +133,10 @@ 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 torchtext torchaudio --force-reinstall --index-url https://download.pytorch.org/whl/nightly/{pt_nightly}" + 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" ) else: self.install_torch_packages(cuda_version)