From b58fc30ed99175995e60055c1875256f72ac5414 Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Wed, 16 Oct 2024 14:04:12 -0700 Subject: [PATCH] Pull 2.5 binaries (#3055) * [DO NOTE MERGE] 2.5 RC TEST --- .jenkins/build.sh | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/.jenkins/build.sh b/.jenkins/build.sh index 8830c4259a..60867ce1d3 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -21,9 +21,10 @@ 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.4 to merge all 2.4 PRs - uncomment to install nightly binaries (update the version as needed). -# pip uninstall -y torch torchvision torchaudio torchtext torchdata -# pip3 install torch==2.4.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124 +# Install 2.5 to merge all 2.4 PRs - uncomment to install nightly binaries (update the version as needed). +pip uninstall -y torch torchvision torchaudio torchtext torchdata +pip3 install torch==2.5.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124 +pip3 install fbgemm-gpu==1.0.0 torchrec==1.0.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124 # Install two language tokenizers for Translation with TorchText tutorial python -m spacy download en_core_web_sm