diff --git a/src/sagemaker/image_uri_config/autogluon.json b/src/sagemaker/image_uri_config/autogluon.json
index 425779a125..2fab95eefc 100644
--- a/src/sagemaker/image_uri_config/autogluon.json
+++ b/src/sagemaker/image_uri_config/autogluon.json
@@ -102,6 +102,7 @@
                     "us-west-2": "763104351884"
                 },
                 "repository": "autogluon-training",
+                "processors": ["cpu", "gpu"],
                 "py_versions": ["py38"]
             },
             "0.4.2": {
@@ -134,6 +135,7 @@
                     "us-west-2": "763104351884"
                 },
                 "repository": "autogluon-training",
+                "processors": ["cpu", "gpu"],
                 "py_versions": ["py38"]
             },
             "0.4.3": {
@@ -166,6 +168,7 @@
                     "us-west-2": "763104351884"
                 },
                 "repository": "autogluon-training",
+                "processors": ["cpu", "gpu"],
                 "py_versions": ["py38"]
             },
             "0.5.2": {
@@ -198,6 +201,7 @@
                     "us-west-2": "763104351884"
                 },
                 "repository": "autogluon-training",
+                "processors": ["cpu", "gpu"],
                 "py_versions": ["py38"]
             },
             "0.6.1": {
@@ -229,6 +233,7 @@
                     "us-west-2": "763104351884"
                 },
                 "repository": "autogluon-training",
+                "processors": ["cpu", "gpu"],
                 "py_versions": ["py38"]
             }
         }