Skip to content

Commit 90a7da8

Browse files
authoredJan 29, 2025··
Move back to use stable binaries 2.6 (#3270)
* Move back to use stable binaries 2.6
1 parent 7216237 commit 90a7da8

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed
 

‎.ci/docker/requirements.txt

+2-2
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ tqdm==4.66.1
1414
numpy==1.24.4
1515
matplotlib
1616
librosa
17-
torch==2.5
17+
torch==2.6
1818
torchvision
1919
torchdata
2020
networkx
@@ -70,4 +70,4 @@ semilearn==0.3.2
7070
torchao==0.5.0
7171
segment_anything==1.0
7272
torchrec==1.0.0; platform_system == "Linux"
73-
fbgemm-gpu==1.0.0; platform_system == "Linux"
73+
fbgemm-gpu==1.1.0; platform_system == "Linux"

‎.jenkins/build.sh

+4-4
Original file line numberDiff line numberDiff line change
@@ -22,10 +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 torch torchvision torchaudio torchtext torchdata
26-
sudo pip3 install torch==2.6.0 torchvision --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124
27-
sudo pip uninstall -y fbgemm-gpu torchrec
28-
sudo pip3 install fbgemm-gpu==1.1.0 torchrec==1.0.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124
25+
# sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata
26+
# sudo pip3 install torch==2.6.0 torchvision --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124
27+
# sudo pip uninstall -y fbgemm-gpu torchrec
28+
# sudo pip3 install fbgemm-gpu==1.1.0 torchrec==1.0.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124
2929

3030
# Install two language tokenizers for Translation with TorchText tutorial
3131
python -m spacy download en_core_web_sm

0 commit comments

Comments
 (0)
Please sign in to comment.