From dfb87240e7cc71bb1c9e8ba53883e4aa4dd10867 Mon Sep 17 00:00:00 2001
From: Fabien Hertschuh <1091026+hertschuh@users.noreply.github.com>
Date: Wed, 22 Oct 2025 10:23:36 -0700
Subject: [PATCH] Update torch version to 2.9.0.

Both for CPU and CUDA.
---
 requirements-torch-cuda.txt | 2 +-
 requirements.txt            | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt
index 5e665660..f221b75a 100644
--- a/requirements-torch-cuda.txt
+++ b/requirements-torch-cuda.txt
@@ -3,7 +3,7 @@ tensorflow-cpu
 
 # Torch with cuda support.
 --extra-index-url https://download.pytorch.org/whl/cu121
-torch==2.7.1
+torch==2.9.0
 
 # Jax cpu-only version.
 jax[cpu]
diff --git a/requirements.txt b/requirements.txt
index 209a378e..d78ae00e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,7 +4,7 @@ tensorflow>=2.20.0;sys_platform == 'darwin'
 
 # Torch.
 --extra-index-url https://download.pytorch.org/whl/cpu
-torch>=2.1.0
+torch>=2.9.0
 
 # Jax.
 jax[cpu]>=0.7.0