From d5dcbbe669c0525350fc677776a2e465bb138270 Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Tue, 28 Jan 2025 14:38:58 -0800 Subject: [PATCH 1/4] Move back to use stable binaries 2.6 --- .ci/docker/requirements.txt | 2 +- .jenkins/build.sh | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt index 56df738f96c..309890ae1ab 100644 --- a/.ci/docker/requirements.txt +++ b/.ci/docker/requirements.txt @@ -70,4 +70,4 @@ semilearn==0.3.2 torchao==0.5.0 segment_anything==1.0 torchrec==1.0.0; platform_system == "Linux" -fbgemm-gpu==1.0.0; platform_system == "Linux" +fbgemm-gpu==1.1.0; platform_system == "Linux" diff --git a/.jenkins/build.sh b/.jenkins/build.sh index 8eca78ae346..4a869d35a75 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -22,10 +22,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.5 to merge all 2.4 PRs - uncomment to install nightly binaries (update the version as needed). -sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata -sudo pip3 install torch==2.6.0 torchvision --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124 -sudo pip uninstall -y fbgemm-gpu torchrec -sudo pip3 install fbgemm-gpu==1.1.0 torchrec==1.0.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124 +# sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata +# sudo pip3 install torch==2.6.0 torchvision --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124 +# sudo pip uninstall -y fbgemm-gpu torchrec +# sudo pip3 install fbgemm-gpu==1.1.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 From 9c9622db8fcb591753636c379c5734e8374e3df3 Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Wed, 29 Jan 2025 08:38:32 -0800 Subject: [PATCH 2/4] Set torch version to 2.6 --- .ci/docker/requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt index 309890ae1ab..c0328356336 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.5 +torch==2.6 torchvision torchdata networkx From f3e80362cf41e165b23102451f351c7f341ee273 Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Wed, 29 Jan 2025 08:58:46 -0800 Subject: [PATCH 3/4] Update --- .ci/docker/requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt index c0328356336..f7052cc1e6b 100644 --- a/.ci/docker/requirements.txt +++ b/.ci/docker/requirements.txt @@ -70,4 +70,4 @@ semilearn==0.3.2 torchao==0.5.0 segment_anything==1.0 torchrec==1.0.0; platform_system == "Linux" -fbgemm-gpu==1.1.0; platform_system == "Linux" +fbgemm-gpu==1.0.0; platform_system == "Linux" From fa609bb4b80a4ccfe45f8b328e1744532553acf1 Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Wed, 29 Jan 2025 10:57:31 -0800 Subject: [PATCH 4/4] Update --- .ci/docker/requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt index f7052cc1e6b..c0328356336 100644 --- a/.ci/docker/requirements.txt +++ b/.ci/docker/requirements.txt @@ -70,4 +70,4 @@ semilearn==0.3.2 torchao==0.5.0 segment_anything==1.0 torchrec==1.0.0; platform_system == "Linux" -fbgemm-gpu==1.0.0; platform_system == "Linux" +fbgemm-gpu==1.1.0; platform_system == "Linux"