Skip to content

Commit

Permalink
Pull 2.6 binaries
Browse files Browse the repository at this point in the history
  • Loading branch information
@svekars
svekars committed Jan 17, 2025
1 parent f7d06b6 commit 86e8e7b
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 4 deletions.
4 changes: 2 additions & 2 deletions .ci/docker/requirements.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -28,8 +28,8 @@ tensorboard
jinja2==3.1.3
pytorch-lightning
torchx
torchrl==0.5.0
tensordict==0.5.0
torchrl==0.6.0
tensordict==0.6.0
ax-platform>=0.4.0
nbformat>=5.9.2
datasets
Expand Down
6 changes: 4 additions & 2 deletions .jenkins/build.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -22,8 +22,10 @@ sudo apt-get install -y pandoc
#Install PyTorch Nightly for test.
# Nightly - pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html
# Install 2.5 to merge all 2.4 PRs - uncomment to install nightly binaries (update the version as needed).
# pip uninstall -y torch torchvision torchaudio torchtext torchdata
# pip3 install torch==2.5.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124
sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata
sudo pip3 install torch==2.6.0 torchvision --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124
sudo pip uninstall -y fbgemm-gpu torchrec
sudo pip3 install fbgemm-gpu==1.1.0 torchrec==1.0.0 --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124

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

0 comments on commit 86e8e7b

Please sign in to comment.