From e5d168a4e5e65eb8bdb1215f416da9dfc60f21da Mon Sep 17 00:00:00 2001 From: agunapal Date: Sat, 9 Mar 2024 02:18:19 +0000 Subject: [PATCH 1/2] torchtext nightly should be cpu --- ts_scripts/install_dependencies.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ts_scripts/install_dependencies.py b/ts_scripts/install_dependencies.py index f32d456f46..0f77b7e46a 100644 --- a/ts_scripts/install_dependencies.py +++ b/ts_scripts/install_dependencies.py @@ -133,8 +133,9 @@ 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) From 20e122c57e6256a71e84dc2e83abe4872215ba0e Mon Sep 17 00:00:00 2001 From: agunapal Date: Sat, 9 Mar 2024 02:37:17 +0000 Subject: [PATCH 2/2] lint failure --- ts_scripts/install_dependencies.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ts_scripts/install_dependencies.py b/ts_scripts/install_dependencies.py index 0f77b7e46a..67f4fad869 100644 --- a/ts_scripts/install_dependencies.py +++ b/ts_scripts/install_dependencies.py @@ -135,7 +135,9 @@ def install_python_packages(self, cuda_version, requirements_file_path, nightly) 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") + os.system( + f"pip3 install --pre torchtext --index-url https://download.pytorch.org/whl/nightly/cpu" + ) else: self.install_torch_packages(cuda_version)