diff --git a/sagemaker-train/tests/data/script_mode/requirements.txt b/sagemaker-train/tests/data/script_mode/requirements.txt
index da7441eee2..ada0b275b2 100644
--- a/sagemaker-train/tests/data/script_mode/requirements.txt
+++ b/sagemaker-train/tests/data/script_mode/requirements.txt
@@ -1,3 +1,3 @@
 numpy
 -f https://download.pytorch.org/whl/torch_stable.html
-torch==2.0.1+cpu
+torch==2.8.0