From 7da5bf6bb6e150c1283c9d1808294eeb45b433f6 Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Mon, 14 Oct 2024 14:44:57 -0700 Subject: [PATCH 1/2] Pin to use 2.5 - Update torch to 2.5 - Update torchrec to use 1.0.0 - Update fbgemm to use 1.0.0 --- .ci/docker/requirements.txt | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt index 01e6a829261..48d88cf0c1b 100644 --- a/.ci/docker/requirements.txt +++ b/.ci/docker/requirements.txt @@ -14,7 +14,7 @@ tqdm==4.66.1 numpy==1.24.4 matplotlib librosa -torch==2.4 +torch==2.5 torchvision torchtext torchdata @@ -70,5 +70,5 @@ pycocotools semilearn==0.3.2 torchao==0.5.0 segment_anything==1.0 -torchrec==0.8.0 -fbgemm-gpu==0.8.0 +torchrec==1.0.0 +fbgemm-gpu==1.0.0 From 8487e9966913a5e3b89cf6df6633e88f2996c8e2 Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Thu, 17 Oct 2024 09:08:25 -0700 Subject: [PATCH 2/2] Undo pip install in build.sh --- .jenkins/build.sh | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/.jenkins/build.sh b/.jenkins/build.sh index 60867ce1d34..e6ecaf6712a 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -22,9 +22,8 @@ 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). -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 +# 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 # Install two language tokenizers for Translation with TorchText tutorial python -m spacy download en_core_web_sm