Skip to content

Commit 27593ea

Browse files
authored
Update torchrl, tensordict
1 parent 5ada2e0 commit 27593ea

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

.jenkins/build.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ sudo apt-get install -y pandoc
2626
sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata torchrl tensordict
2727
# sudo pip3 install fbgemm-gpu==1.1.0 torchrec==1.0.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124
2828
pip3 install torch==2.8.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126
29+
pip3 install torchrl==0.9.1 tensordict==0.9.1
2930
# Install two language tokenizers for Translation with TorchText tutorial
3031
python -m spacy download en_core_web_sm
3132
python -m spacy download de_core_news_sm

0 commit comments

Comments
 (0)