diff --git a/sdks/python/container/ml/py310/base_image_requirements.txt b/sdks/python/container/ml/py310/base_image_requirements.txt index 415b7579dc8b..660e9a62af22 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 a1ece225e0c6..1bd0f9d6f2a7 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