diff --git a/sdks/python/container/ml/py312/base_image_requirements.txt b/sdks/python/container/ml/py312/base_image_requirements.txt index 2b8283180794..15ac8acc92fd 100644 --- a/sdks/python/container/ml/py312/base_image_requirements.txt +++ b/sdks/python/container/ml/py312/base_image_requirements.txt @@ -218,7 +218,7 @@ testcontainers==4.14.2 textual==8.2.7 threadpoolctl==3.6.0 tokenizers==0.21.4 -torch==2.8.0+cpu +torch==2.12.0 tqdm==4.68.2 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 c659def93a8c..46e65ea542a2 100644 --- a/sdks/python/container/ml/py312/gpu_image_requirements.txt +++ b/sdks/python/container/ml/py312/gpu_image_requirements.txt @@ -308,7 +308,7 @@ textual==8.2.7 threadpoolctl==3.6.0 tiktoken==0.13.0 tokenizers==0.21.4 -torch==2.7.1 +torch==2.12.0 torchaudio==2.7.1 torchvision==0.22.1 tqdm==4.68.2