diff --git a/sdks/python/container/ml/py312/base_image_requirements.txt b/sdks/python/container/ml/py312/base_image_requirements.txt
index c72558f8acd9..ebab1e675c70 100644
--- a/sdks/python/container/ml/py312/base_image_requirements.txt
+++ b/sdks/python/container/ml/py312/base_image_requirements.txt
@@ -209,7 +209,7 @@ termcolor==3.3.0
 testcontainers==4.14.2
 threadpoolctl==3.6.0
 tokenizers==0.21.4
-torch==2.8.0+cpu
+torch==2.8.0
 tqdm==4.67.3
 transformers==4.55.4
 typing-inspection==0.4.2
diff --git a/sdks/python/container/ml/py312/gpu_image_requirements.txt b/sdks/python/container/ml/py312/gpu_image_requirements.txt
index 632ba689e3ee..eb4c6d6577c1 100644
--- a/sdks/python/container/ml/py312/gpu_image_requirements.txt
+++ b/sdks/python/container/ml/py312/gpu_image_requirements.txt
@@ -284,7 +284,7 @@ testcontainers==4.14.1
 threadpoolctl==3.6.0
 tiktoken==0.12.0
 tokenizers==0.21.4
-torch==2.7.1
+torch==2.8.0
 torchaudio==2.7.1
 torchvision==0.22.1
 tqdm==4.67.3