Added processors for training images for consistency

aws · Dec 20, 2022 · 6861e5b · 6861e5b
1 parent b5e26fe
commit 6861e5b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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"]
             }
         }