From 6861e5be3ede59034fb58e904f91de27ace53a34 Mon Sep 17 00:00:00 2001 From: Alexander Shirkov Date: Mon, 19 Dec 2022 13:46:58 -0800 Subject: [PATCH] Added processors for training images for consistency --- src/sagemaker/image_uri_config/autogluon.json | 5 +++++ 1 file changed, 5 insertions(+) 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"] } }