diff --git a/src/sagemaker/image_uri_config/pytorch-smp.json b/src/sagemaker/image_uri_config/pytorch-smp.json index ab1398666b..3a119c81d2 100644 --- a/src/sagemaker/image_uri_config/pytorch-smp.json +++ b/src/sagemaker/image_uri_config/pytorch-smp.json @@ -8,7 +8,7 @@ "2.1": "2.1.2", "2.2": "2.3.1", "2.2.0": "2.3.1", - "2.3.1": "2.4.0" + "2.3.1": "2.5.0" }, "versions": { "2.0.1": { @@ -136,7 +136,7 @@ }, "repository": "smdistributed-modelparallel" }, - "2.4.0": { + "2.5.0": { "py_versions": [ "py311" ],