Skip to content

Commit 81efd5f

Browse files
authored
Update .jenkins/build.sh
1 parent d463fd4 commit 81efd5f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

.jenkins/build.sh

+2
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,8 @@ sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata torchrl t
2626
pip3 install torch==2.7.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126
2727
sudo pip uninstall -y fbgemm-gpu
2828
sudo pip3 install --pre fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cu126/
29+
pip install tensordict-nightly
30+
pip install torchrl-nightly
2931
#sudo pip3 install fbgemm-gpu==1.1.0 torchrec==1.0.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126
3032

3133

0 commit comments

Comments
 (0)