Skip to content

Commit 3cdb01f

Browse files
authoredMar 21, 2025··

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed
 

‎.jenkins/build.sh

+4-1
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,11 @@ 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
25+
sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata torchrl tensordict
2626
pip3 install torch==2.7.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126
27+
pip install git+https://github.com/pytorch/tensordict
28+
pip install git+https://github.com/pytorch/torchrl
29+
2730
#sudo pip uninstall -y fbgemm-gpu torchrec
2831
#sudo pip3 install fbgemm-gpu==1.1.0 torchrec==1.0.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126
2932

0 commit comments

Comments
 (0)
Please sign in to comment.