diff --git a/.jenkins/build.sh b/.jenkins/build.sh index 9162152ff7..49c1856fc3 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -22,10 +22,13 @@ sudo apt-get install -y pandoc #Install PyTorch Nightly for test. # Nightly - pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html # Install 2.5 to merge all 2.4 PRs - uncomment to install nightly binaries (update the version as needed). -# sudo pip uninstall -y fbgemm-gpu torchrec -# sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata torchrl tensordict -# sudo pip3 install fbgemm-gpu==1.1.0 torchrec==1.0.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124 -# pip3 install torch==2.7.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126 +#sudo pip uninstall -y fbgemm-gpu torchrec +sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata torchrl tensordict +pip3 install torch==2.8.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126 +pip3 install torchrl==0.9.1 tensordict==0.9.1 +# sudo pip3 install fbgemm-gpu==1.2.0 torchrec==1.2.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126 + + # Install two language tokenizers for Translation with TorchText tutorial python -m spacy download en_core_web_sm python -m spacy download de_core_news_sm