diff --git a/sdks/python/container/ml/py310/base_image_requirements.txt b/sdks/python/container/ml/py310/base_image_requirements.txt
index 85a199d9b2a4..ad1ee76c5a82 100644
--- a/sdks/python/container/ml/py310/base_image_requirements.txt
+++ b/sdks/python/container/ml/py310/base_image_requirements.txt
@@ -223,7 +223,7 @@ textual==8.2.7
 threadpoolctl==3.6.0
 tokenizers==0.21.4
 tomli==2.4.1
-torch==2.8.0+cpu
+torch==2.12.1
 tqdm==4.68.3
 transformers==4.55.4
 typing-inspection==0.4.2
diff --git a/sdks/python/container/ml/py310/gpu_image_requirements.txt b/sdks/python/container/ml/py310/gpu_image_requirements.txt
index 9121ebe54150..58aa928ab110 100644
--- a/sdks/python/container/ml/py310/gpu_image_requirements.txt
+++ b/sdks/python/container/ml/py310/gpu_image_requirements.txt
@@ -313,7 +313,7 @@ threadpoolctl==3.6.0
 tiktoken==0.13.0
 tokenizers==0.21.4
 tomli==2.4.1
-torch==2.7.1
+torch==2.12.1
 torchaudio==2.7.1
 torchvision==0.22.1
 tqdm==4.68.3