Merge branch 'main' into thiagofc/add-onnx-registry-tutorial

Author: Svetlana Karslioglu

.jenkins/build.sh

@@ -25,9 +25,9 @@ pip install --progress-bar off -r $DIR/../requirements.txt
 #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.1 for testing
-pip uninstall -y torch torchvision torchaudio torchtext torchdata
-pip3 install torch torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu121
-pip3 install torchdata torchtext --index-url https://download.pytorch.org/whl/test/cpu
+# pip uninstall -y torch torchvision torchaudio torchtext torchdata
+# pip3 install torch torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/test/cu121
+# pip3 install torchdata torchtext --index-url https://download.pytorch.org/whl/test/cpu
 
 # Install two language tokenizers for Translation with TorchText tutorial
 python -m spacy download en_core_web_sm

requirements.txt

@@ -25,7 +25,7 @@ tensorboard
 jinja2==3.0.3
 pytorch-lightning
 torchx
-torchrl
+torchrl==0.2.0
 ax-platform
 nbformat>=4.2.0
 datasets