diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt index 7aede51dda..a355966683 100644 --- a/.ci/docker/requirements.txt +++ b/.ci/docker/requirements.txt @@ -13,7 +13,7 @@ tqdm==4.66.1 numpy==1.24.4 matplotlib librosa -torch==2.3 +torch==2.4 torchvision torchtext torchdata diff --git a/.jenkins/build.sh b/.jenkins/build.sh index dbec2d2552..8830c4259a 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -22,8 +22,8 @@ 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.4 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.4.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu124 +# pip uninstall -y torch torchvision torchaudio torchtext torchdata +# pip3 install torch==2.4.0 torchvision torchaudio --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