From 022d48ecc4921c79757198c3a64a92618c842b92 Mon Sep 17 00:00:00 2001
From: sekyondaMeta <127536312+sekyondaMeta@users.noreply.github.com>
Date: Thu, 31 Jul 2025 10:26:36 -0400
Subject: [PATCH 1/2] Update requirements.txt to torch2_8

Updating to torch2_8
---
 .ci/docker/requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt
index f1f105122ed..e9710f5d37f 100644
--- a/.ci/docker/requirements.txt
+++ b/.ci/docker/requirements.txt
@@ -24,7 +24,7 @@ tqdm==4.66.1
 numpy==1.24.4
 matplotlib
 librosa
-torch==2.7
+torch==2.8
 torchvision
 torchdata
 networkx

From 386eef7b90adcd05676f49a582dc25e6dd562384 Mon Sep 17 00:00:00 2001
From: sekyondaMeta <127536312+sekyondaMeta@users.noreply.github.com>
Date: Thu, 31 Jul 2025 10:28:45 -0400
Subject: [PATCH 2/2] Update requirements.txt

---
 .ci/docker/requirements.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt
index e9710f5d37f..1f0391c6223 100644
--- a/.ci/docker/requirements.txt
+++ b/.ci/docker/requirements.txt
@@ -38,8 +38,8 @@ tensorboard
 jinja2==3.1.3
 pytorch-lightning
 torchx
-torchrl==0.7.2
-tensordict==0.7.2
+torchrl==0.9.2
+tensordict==0.9.1
 # For ax_multiobjective_nas_tutorial.py
 ax-platform>=0.4.0,<0.5.0
 nbformat>=5.9.2