diff --git a/src/sagemaker/image_uri_config/pytorch-smp.json b/src/sagemaker/image_uri_config/pytorch-smp.json
index 18eb0fb4c3..91d8ab9184 100644
--- a/src/sagemaker/image_uri_config/pytorch-smp.json
+++ b/src/sagemaker/image_uri_config/pytorch-smp.json
@@ -9,7 +9,7 @@
             "2.2": "2.3.1",
             "2.2.0": "2.3.1",
             "2.3.1": "2.5.0",
-            "2.4.1": "2.6.0"
+            "2.4.1": "2.6.1"
         },
         "versions": {
             "2.0.1": {
@@ -162,7 +162,7 @@
                 },
                 "repository": "smdistributed-modelparallel"
             },
-            "2.6.0": {
+            "2.6.1": {
                 "py_versions": [
                     "py311"
                 ],