diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_cu11.8_nightly.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_cu11.8_nightly.txt
index a90eade..96d4c0d 100644
--- a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_cu11.8_nightly.txt
+++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_cu11.8_nightly.txt
@@ -1,6 +1,6 @@
 --pre
 -f https://download.pytorch.org/whl/torch_stable.html
-torch==2.0.0+cu118
+torch==2.8.0
 torchvision
 torchtext
 --pre
diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_cu11.8_stable.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_cu11.8_stable.txt
index 316b3d2..16fb28a 100644
--- a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_cu11.8_stable.txt
+++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_cu11.8_stable.txt
@@ -1,6 +1,6 @@
 --pre
 -f https://download.pytorch.org/whl/torch_stable.html
-torch==2.0.0+cu118
+torch==2.8.0
 torchvision
 torchtext
 --pre