diff --git a/images/requirements.txt b/images/requirements.txt
index 2b4e980..d3dc8ae 100644
--- a/images/requirements.txt
+++ b/images/requirements.txt
@@ -130,7 +130,7 @@ tensorflow-io-gcs-filesystem==0.30.0
 termcolor==2.2.0
 toml==0.10.2
 tomli==2.0.1
-torch==1.13.1
+torch==2.8.0
 torchmetrics==0.11.0
 torchrec==0.3.2
 torchsnapshot==0.1.0