Skip to content

Commit c097493

Browse files
committed
Update pin fbgemm-gpu to 1.1
1 parent 0b4ff0c commit c097493

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

.jenkins/build.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@ sudo apt-get install -y pandoc
2424
# Install 2.5 to merge all 2.4 PRs - uncomment to install nightly binaries (update the version as needed).
2525
sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata
2626
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
2729

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

0 commit comments

Comments
 (0)