diff --git a/ts_scripts/install_dependencies.py b/ts_scripts/install_dependencies.py index ff9ae4c25e..3296844482 100644 --- a/ts_scripts/install_dependencies.py +++ b/ts_scripts/install_dependencies.py @@ -131,7 +131,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 torchtext torchaudio --force-reinstall --extra-index-url https://download.pytorch.org/whl/nightly/{pt_nightly}" + f"pip3 install numpy --pre torch torchvision torchtext torchaudio --force-reinstall --index-url https://download.pytorch.org/whl/nightly/{pt_nightly}" ) else: self.install_torch_packages(cuda_version)