diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt
index 5b1eca2c89..30f22b5d55 100644
--- a/requirements-torch-cuda.txt
+++ b/requirements-torch-cuda.txt
@@ -3,8 +3,8 @@ tensorflow-cpu>=2.16.1
 
 # Torch with cuda support.
 --extra-index-url https://download.pytorch.org/whl/cu121
-torch==2.4.0+cu121
-torchvision==0.19.0+cu121
+torch==2.6.0+cpu
+torchvision==0.21.0+cpu
 
 # Jax cpu-only version.
 jax[cpu]