diff --git a/src/sagemaker/fw_utils.py b/src/sagemaker/fw_utils.py index f5547b50bd..8a644eb305 100644 --- a/src/sagemaker/fw_utils.py +++ b/src/sagemaker/fw_utils.py @@ -79,6 +79,7 @@ "ml.p3.16xlarge", "ml.p3dn.24xlarge", "ml.p4d.24xlarge", + "ml.p4de.24xlarge", "local_gpu", ) SM_DATAPARALLEL_SUPPORTED_FRAMEWORK_VERSIONS = {