Skip to content

Commit 616c1ad

Browse files
authored
Update pytorch nightly installation to use cpu version of torchtext (#3011)
* torchtext nightly should be cpu * lint failure
1 parent b92cf42 commit 616c1ad

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

ts_scripts/install_dependencies.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,10 @@ def install_python_packages(self, cuda_version, requirements_file_path, nightly)
133133
if nightly:
134134
pt_nightly = "cpu" if not cuda_version else cuda_version
135135
os.system(
136-
f"pip3 install numpy --pre torch torchvision torchtext torchaudio --force-reinstall --index-url https://download.pytorch.org/whl/nightly/{pt_nightly}"
136+
f"pip3 install numpy --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/{pt_nightly}"
137+
)
138+
os.system(
139+
f"pip3 install --pre torchtext --index-url https://download.pytorch.org/whl/nightly/cpu"
137140
)
138141
else:
139142
self.install_torch_packages(cuda_version)

0 commit comments

Comments
 (0)