diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt index 6babb63e78..56df738f96 100644 --- a/.ci/docker/requirements.txt +++ b/.ci/docker/requirements.txt @@ -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 diff --git a/.jenkins/build.sh b/.jenkins/build.sh index e6ecaf6712..8eca78ae34 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -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 diff --git a/.jenkins/validate_tutorials_built.py b/.jenkins/validate_tutorials_built.py index c3bf4c5534..665b3b48e3 100644 --- a/.jenkins/validate_tutorials_built.py +++ b/.jenkins/validate_tutorials_built.py @@ -51,7 +51,10 @@ "intermediate_source/flask_rest_api_tutorial", "intermediate_source/text_to_speech_with_torchaudio", "intermediate_source/tensorboard_profiler_tutorial", # reenable after 2.0 release. - "intermediate_source/torch_export_tutorial" # reenable after 2940 is fixed. + "intermediate_source/torch_export_tutorial", # reenable after 2940 is fixed. + "advanced_source/pendulum", + "beginner_source/onnx/export_simple_model_to_onnx_tutorial", + "beginner_source/onnx/onnx_registry_tutorial" ] def tutorial_source_dirs() -> List[Path]: