diff --git a/sdks/python/container/ml/py311/base_image_requirements.txt b/sdks/python/container/ml/py311/base_image_requirements.txt index 3f9a18099fa2..9b7e62be4412 100644 --- a/sdks/python/container/ml/py311/base_image_requirements.txt +++ b/sdks/python/container/ml/py311/base_image_requirements.txt @@ -219,7 +219,7 @@ testcontainers==4.14.2 textual==8.2.7 threadpoolctl==3.6.0 tokenizers==0.21.4 -torch==2.8.0+cpu +torch==2.13.0 tqdm==4.68.3 transformers==4.55.4 typing-inspection==0.4.2 diff --git a/sdks/python/container/ml/py311/gpu_image_requirements.txt b/sdks/python/container/ml/py311/gpu_image_requirements.txt index 7cb148c46f98..da8d359efb70 100644 --- a/sdks/python/container/ml/py311/gpu_image_requirements.txt +++ b/sdks/python/container/ml/py311/gpu_image_requirements.txt @@ -309,7 +309,7 @@ textual==8.2.7 threadpoolctl==3.6.0 tiktoken==0.13.0 tokenizers==0.21.4 -torch==2.7.1 +torch==2.13.0 torchaudio==2.7.1 torchvision==0.22.1 tqdm==4.68.3