From 39617da75a00df031696e0914074811dfe10b48d Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Wed, 16 Apr 2025 14:06:31 -0700 Subject: [PATCH] Pull 2.7 binaries --- .ci/docker/requirements.txt | 4 ++-- .jenkins/build.sh | 4 +++- .jenkins/validate_tutorials_built.py | 9 ++++++++- 3 files changed, 13 insertions(+), 4 deletions(-) diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt index 89dd788ae7..0e95c62c6b 100644 --- a/.ci/docker/requirements.txt +++ b/.ci/docker/requirements.txt @@ -36,7 +36,7 @@ datasets transformers torchmultimodal-nightly # needs to be updated to stable as soon as it's avaialable onnx -onnxscript +onnxscript>=0.2.2 onnxruntime evaluate accelerate>=0.20.1 @@ -69,5 +69,5 @@ pycocotools semilearn==0.3.2 torchao==0.5.0 segment_anything==1.0 -torchrec==1.0.0; platform_system == "Linux" +torchrec==1.1.0; platform_system == "Linux" fbgemm-gpu==1.1.0; platform_system == "Linux" diff --git a/.jenkins/build.sh b/.jenkins/build.sh index 4a869d35a7..8786859d7d 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -26,7 +26,9 @@ sudo apt-get install -y pandoc # 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 - +sudo pip uninstall -y torch torchvision torchaudio torchtext torchdata torchrl tensordict +pip3 install torch==2.7.0 torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu126 +#sudo pip uninstall -y fbgemm-gpu # Install two language tokenizers for Translation with TorchText tutorial python -m spacy download en_core_web_sm python -m spacy download de_core_news_sm diff --git a/.jenkins/validate_tutorials_built.py b/.jenkins/validate_tutorials_built.py index 984632156e..3ed1e0c028 100644 --- a/.jenkins/validate_tutorials_built.py +++ b/.jenkins/validate_tutorials_built.py @@ -51,7 +51,14 @@ "intermediate_source/text_to_speech_with_torchaudio", "intermediate_source/tensorboard_profiler_tutorial", # reenable after 2.0 release. "advanced_source/semi_structured_sparse", # reenable after 3303 is fixed. - "recipes_source/recipes/reasoning_about_shapes" + "intermediate_source/mario_rl_tutorial", # reenable after 3302 is fixed + "intermediate_source/reinforcement_ppo", # reenable after 3302 is fixed + "intermediate_source/pinmem_nonblock", # reenable after 3302 is fixed + "intermediate_source/dqn_with_rnn_tutorial", # reenable after 3302 is fixed + "advanced_source/pendulum", # reenable after 3302 is fixed + "advanced_source/coding_ddpg", # reenable after 3302 is fixed + "intermediate_source/torchrec_intro_tutorial", # reenable after 3302 is fixed + "recipes_source/recipes/reasoning_about_shapes" # reenable after 3326 is fixed ] def tutorial_source_dirs() -> List[Path]: