diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt
index 43dc4c5ef5..050dd85b1c 100644
--- a/requirements-torch-cuda.txt
+++ b/requirements-torch-cuda.txt
@@ -4,8 +4,8 @@ tensorflow-text~=2.16.1
 
 # Torch with cuda support.
 --extra-index-url https://download.pytorch.org/whl/cu121
-torch==2.1.2
-torchvision==0.16.2
+torch==2.2.1+cu121
+torchvision==0.17.1+cu121
 
 # Jax cpu-only version.
 jax[cpu]