Skip to content

Commit f62e4d3

Browse files
authored
Update build.sh
1 parent 3825480 commit f62e4d3

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

.jenkins/build.sh

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -22,13 +22,10 @@ sudo apt-get install -y pandoc
2222
#Install PyTorch Nightly for test.
2323
# Nightly - pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html
2424
# Install 2.5 to merge all 2.4 PRs - uncomment to install nightly binaries (update the version as needed).
25-
#sudo pip uninstall -y fbgemm-gpu torchrec
26-
sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata torchrl tensordict
27-
pip3 install torch==2.8.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126
28-
pip3 install torchrl==0.9.1 tensordict==0.9.1
29-
# sudo pip3 install fbgemm-gpu==1.2.0 torchrec==1.2.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126
30-
31-
25+
# sudo pip uninstall -y fbgemm-gpu torchrec
26+
# sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata torchrl tensordict
27+
# sudo pip3 install fbgemm-gpu==1.1.0 torchrec==1.0.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124
28+
# pip3 install torch==2.7.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126
3229
# Install two language tokenizers for Translation with TorchText tutorial
3330
python -m spacy download en_core_web_sm
3431
python -m spacy download de_core_news_sm

0 commit comments

Comments
 (0)