Amazon SageMaker Service

2025/08/27 - Amazon SageMaker Service - 3 updated api methods

Changes  This release adds support for AutoScaling on SageMaker HyperPod.

CreateCluster (updated) Link ¶
Changes (request)
{'AutoScaling': {'AutoScalerType': 'Karpenter', 'Mode': 'Enable | Disable'},
 'ClusterRole': 'string'}

Creates a SageMaker HyperPod cluster. SageMaker HyperPod is a capability of SageMaker for creating and managing persistent clusters for developing large machine learning models, such as large language models (LLMs) and diffusion models. To learn more, see Amazon SageMaker HyperPod in the Amazon SageMaker Developer Guide.

See also: AWS API Documentation

Request Syntax

client.create_cluster(
    ClusterName='string',
    InstanceGroups=[
        {
            'InstanceCount': 123,
            'InstanceGroupName': 'string',
            'InstanceType': 'ml.p4d.24xlarge'|'ml.p4de.24xlarge'|'ml.p5.48xlarge'|'ml.p6e-gb200.36xlarge'|'ml.trn1.32xlarge'|'ml.trn1n.32xlarge'|'ml.g5.xlarge'|'ml.g5.2xlarge'|'ml.g5.4xlarge'|'ml.g5.8xlarge'|'ml.g5.12xlarge'|'ml.g5.16xlarge'|'ml.g5.24xlarge'|'ml.g5.48xlarge'|'ml.c5.large'|'ml.c5.xlarge'|'ml.c5.2xlarge'|'ml.c5.4xlarge'|'ml.c5.9xlarge'|'ml.c5.12xlarge'|'ml.c5.18xlarge'|'ml.c5.24xlarge'|'ml.c5n.large'|'ml.c5n.2xlarge'|'ml.c5n.4xlarge'|'ml.c5n.9xlarge'|'ml.c5n.18xlarge'|'ml.m5.large'|'ml.m5.xlarge'|'ml.m5.2xlarge'|'ml.m5.4xlarge'|'ml.m5.8xlarge'|'ml.m5.12xlarge'|'ml.m5.16xlarge'|'ml.m5.24xlarge'|'ml.t3.medium'|'ml.t3.large'|'ml.t3.xlarge'|'ml.t3.2xlarge'|'ml.g6.xlarge'|'ml.g6.2xlarge'|'ml.g6.4xlarge'|'ml.g6.8xlarge'|'ml.g6.16xlarge'|'ml.g6.12xlarge'|'ml.g6.24xlarge'|'ml.g6.48xlarge'|'ml.gr6.4xlarge'|'ml.gr6.8xlarge'|'ml.g6e.xlarge'|'ml.g6e.2xlarge'|'ml.g6e.4xlarge'|'ml.g6e.8xlarge'|'ml.g6e.16xlarge'|'ml.g6e.12xlarge'|'ml.g6e.24xlarge'|'ml.g6e.48xlarge'|'ml.p5e.48xlarge'|'ml.p5en.48xlarge'|'ml.p6-b200.48xlarge'|'ml.trn2.48xlarge'|'ml.c6i.large'|'ml.c6i.xlarge'|'ml.c6i.2xlarge'|'ml.c6i.4xlarge'|'ml.c6i.8xlarge'|'ml.c6i.12xlarge'|'ml.c6i.16xlarge'|'ml.c6i.24xlarge'|'ml.c6i.32xlarge'|'ml.m6i.large'|'ml.m6i.xlarge'|'ml.m6i.2xlarge'|'ml.m6i.4xlarge'|'ml.m6i.8xlarge'|'ml.m6i.12xlarge'|'ml.m6i.16xlarge'|'ml.m6i.24xlarge'|'ml.m6i.32xlarge'|'ml.r6i.large'|'ml.r6i.xlarge'|'ml.r6i.2xlarge'|'ml.r6i.4xlarge'|'ml.r6i.8xlarge'|'ml.r6i.12xlarge'|'ml.r6i.16xlarge'|'ml.r6i.24xlarge'|'ml.r6i.32xlarge'|'ml.i3en.large'|'ml.i3en.xlarge'|'ml.i3en.2xlarge'|'ml.i3en.3xlarge'|'ml.i3en.6xlarge'|'ml.i3en.12xlarge'|'ml.i3en.24xlarge'|'ml.m7i.large'|'ml.m7i.xlarge'|'ml.m7i.2xlarge'|'ml.m7i.4xlarge'|'ml.m7i.8xlarge'|'ml.m7i.12xlarge'|'ml.m7i.16xlarge'|'ml.m7i.24xlarge'|'ml.m7i.48xlarge'|'ml.r7i.large'|'ml.r7i.xlarge'|'ml.r7i.2xlarge'|'ml.r7i.4xlarge'|'ml.r7i.8xlarge'|'ml.r7i.12xlarge'|'ml.r7i.16xlarge'|'ml.r7i.24xlarge'|'ml.r7i.48xlarge',
            'LifeCycleConfig': {
                'SourceS3Uri': 'string',
                'OnCreate': 'string'
            },
            'ExecutionRole': 'string',
            'ThreadsPerCore': 123,
            'InstanceStorageConfigs': [
                {
                    'EbsVolumeConfig': {
                        'VolumeSizeInGB': 123,
                        'VolumeKmsKeyId': 'string',
                        'RootVolume': True|False
                    }
                },
            ],
            'OnStartDeepHealthChecks': [
                'InstanceStress'|'InstanceConnectivity',
            ],
            'TrainingPlanArn': 'string',
            'OverrideVpcConfig': {
                'SecurityGroupIds': [
                    'string',
                ],
                'Subnets': [
                    'string',
                ]
            },
            'ScheduledUpdateConfig': {
                'ScheduleExpression': 'string',
                'DeploymentConfig': {
                    'RollingUpdatePolicy': {
                        'MaximumBatchSize': {
                            'Type': 'INSTANCE_COUNT'|'CAPACITY_PERCENTAGE',
                            'Value': 123
                        },
                        'RollbackMaximumBatchSize': {
                            'Type': 'INSTANCE_COUNT'|'CAPACITY_PERCENTAGE',
                            'Value': 123
                        }
                    },
                    'WaitIntervalInSeconds': 123,
                    'AutoRollbackConfiguration': [
                        {
                            'AlarmName': 'string'
                        },
                    ]
                }
            },
            'ImageId': 'string'
        },
    ],
    RestrictedInstanceGroups=[
        {
            'InstanceCount': 123,
            'InstanceGroupName': 'string',
            'InstanceType': 'ml.p4d.24xlarge'|'ml.p4de.24xlarge'|'ml.p5.48xlarge'|'ml.p6e-gb200.36xlarge'|'ml.trn1.32xlarge'|'ml.trn1n.32xlarge'|'ml.g5.xlarge'|'ml.g5.2xlarge'|'ml.g5.4xlarge'|'ml.g5.8xlarge'|'ml.g5.12xlarge'|'ml.g5.16xlarge'|'ml.g5.24xlarge'|'ml.g5.48xlarge'|'ml.c5.large'|'ml.c5.xlarge'|'ml.c5.2xlarge'|'ml.c5.4xlarge'|'ml.c5.9xlarge'|'ml.c5.12xlarge'|'ml.c5.18xlarge'|'ml.c5.24xlarge'|'ml.c5n.large'|'ml.c5n.2xlarge'|'ml.c5n.4xlarge'|'ml.c5n.9xlarge'|'ml.c5n.18xlarge'|'ml.m5.large'|'ml.m5.xlarge'|'ml.m5.2xlarge'|'ml.m5.4xlarge'|'ml.m5.8xlarge'|'ml.m5.12xlarge'|'ml.m5.16xlarge'|'ml.m5.24xlarge'|'ml.t3.medium'|'ml.t3.large'|'ml.t3.xlarge'|'ml.t3.2xlarge'|'ml.g6.xlarge'|'ml.g6.2xlarge'|'ml.g6.4xlarge'|'ml.g6.8xlarge'|'ml.g6.16xlarge'|'ml.g6.12xlarge'|'ml.g6.24xlarge'|'ml.g6.48xlarge'|'ml.gr6.4xlarge'|'ml.gr6.8xlarge'|'ml.g6e.xlarge'|'ml.g6e.2xlarge'|'ml.g6e.4xlarge'|'ml.g6e.8xlarge'|'ml.g6e.16xlarge'|'ml.g6e.12xlarge'|'ml.g6e.24xlarge'|'ml.g6e.48xlarge'|'ml.p5e.48xlarge'|'ml.p5en.48xlarge'|'ml.p6-b200.48xlarge'|'ml.trn2.48xlarge'|'ml.c6i.large'|'ml.c6i.xlarge'|'ml.c6i.2xlarge'|'ml.c6i.4xlarge'|'ml.c6i.8xlarge'|'ml.c6i.12xlarge'|'ml.c6i.16xlarge'|'ml.c6i.24xlarge'|'ml.c6i.32xlarge'|'ml.m6i.large'|'ml.m6i.xlarge'|'ml.m6i.2xlarge'|'ml.m6i.4xlarge'|'ml.m6i.8xlarge'|'ml.m6i.12xlarge'|'ml.m6i.16xlarge'|'ml.m6i.24xlarge'|'ml.m6i.32xlarge'|'ml.r6i.large'|'ml.r6i.xlarge'|'ml.r6i.2xlarge'|'ml.r6i.4xlarge'|'ml.r6i.8xlarge'|'ml.r6i.12xlarge'|'ml.r6i.16xlarge'|'ml.r6i.24xlarge'|'ml.r6i.32xlarge'|'ml.i3en.large'|'ml.i3en.xlarge'|'ml.i3en.2xlarge'|'ml.i3en.3xlarge'|'ml.i3en.6xlarge'|'ml.i3en.12xlarge'|'ml.i3en.24xlarge'|'ml.m7i.large'|'ml.m7i.xlarge'|'ml.m7i.2xlarge'|'ml.m7i.4xlarge'|'ml.m7i.8xlarge'|'ml.m7i.12xlarge'|'ml.m7i.16xlarge'|'ml.m7i.24xlarge'|'ml.m7i.48xlarge'|'ml.r7i.large'|'ml.r7i.xlarge'|'ml.r7i.2xlarge'|'ml.r7i.4xlarge'|'ml.r7i.8xlarge'|'ml.r7i.12xlarge'|'ml.r7i.16xlarge'|'ml.r7i.24xlarge'|'ml.r7i.48xlarge',
            'ExecutionRole': 'string',
            'ThreadsPerCore': 123,
            'InstanceStorageConfigs': [
                {
                    'EbsVolumeConfig': {
                        'VolumeSizeInGB': 123,
                        'VolumeKmsKeyId': 'string',
                        'RootVolume': True|False
                    }
                },
            ],
            'OnStartDeepHealthChecks': [
                'InstanceStress'|'InstanceConnectivity',
            ],
            'TrainingPlanArn': 'string',
            'OverrideVpcConfig': {
                'SecurityGroupIds': [
                    'string',
                ],
                'Subnets': [
                    'string',
                ]
            },
            'ScheduledUpdateConfig': {
                'ScheduleExpression': 'string',
                'DeploymentConfig': {
                    'RollingUpdatePolicy': {
                        'MaximumBatchSize': {
                            'Type': 'INSTANCE_COUNT'|'CAPACITY_PERCENTAGE',
                            'Value': 123
                        },
                        'RollbackMaximumBatchSize': {
                            'Type': 'INSTANCE_COUNT'|'CAPACITY_PERCENTAGE',
                            'Value': 123
                        }
                    },
                    'WaitIntervalInSeconds': 123,
                    'AutoRollbackConfiguration': [
                        {
                            'AlarmName': 'string'
                        },
                    ]
                }
            },
            'EnvironmentConfig': {
                'FSxLustreConfig': {
                    'SizeInGiB': 123,
                    'PerUnitStorageThroughput': 123
                }
            }
        },
    ],
    VpcConfig={
        'SecurityGroupIds': [
            'string',
        ],
        'Subnets': [
            'string',
        ]
    },
    Tags=[
        {
            'Key': 'string',
            'Value': 'string'
        },
    ],
    Orchestrator={
        'Eks': {
            'ClusterArn': 'string'
        }
    },
    NodeRecovery='Automatic'|'None',
    NodeProvisioningMode='Continuous',
    ClusterRole='string',
    AutoScaling={
        'Mode': 'Enable'|'Disable',
        'AutoScalerType': 'Karpenter'
    }
)
type ClusterName:

string

param ClusterName:

[REQUIRED]

The name for the new SageMaker HyperPod cluster.

type InstanceGroups:

list

param InstanceGroups:

The instance groups to be created in the SageMaker HyperPod cluster.

  • (dict) --

    The specifications of an instance group that you need to define.

    • InstanceCount (integer) -- [REQUIRED]

      Specifies the number of instances to add to the instance group of a SageMaker HyperPod cluster.

    • InstanceGroupName (string) -- [REQUIRED]

      Specifies the name of the instance group.

    • InstanceType (string) -- [REQUIRED]

      Specifies the instance type of the instance group.

    • LifeCycleConfig (dict) -- [REQUIRED]

      Specifies the LifeCycle configuration for the instance group.

      • SourceS3Uri (string) -- [REQUIRED]

        An Amazon S3 bucket path where your lifecycle scripts are stored.

      • OnCreate (string) -- [REQUIRED]

        The file name of the entrypoint script of lifecycle scripts under SourceS3Uri. This entrypoint script runs during cluster creation.

    • ExecutionRole (string) -- [REQUIRED]

      Specifies an IAM execution role to be assumed by the instance group.

    • ThreadsPerCore (integer) --

      Specifies the value for Threads per core. For instance types that support multithreading, you can specify 1 for disabling multithreading and 2 for enabling multithreading. For instance types that doesn't support multithreading, specify 1. For more information, see the reference table of CPU cores and threads per CPU core per instance type in the Amazon Elastic Compute Cloud User Guide.

    • InstanceStorageConfigs (list) --

      Specifies the additional storage configurations for the instances in the SageMaker HyperPod cluster instance group.

      • (dict) --

        Defines the configuration for attaching additional storage to the instances in the SageMaker HyperPod cluster instance group. To learn more, see SageMaker HyperPod release notes: June 20, 2024.

        • EbsVolumeConfig (dict) --

          Defines the configuration for attaching additional Amazon Elastic Block Store (EBS) volumes to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to /opt/sagemaker.

          • VolumeSizeInGB (integer) --

            The size in gigabytes (GB) of the additional EBS volume to be attached to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to /opt/sagemaker.

          • VolumeKmsKeyId (string) --

            The ID of a KMS key to encrypt the Amazon EBS volume.

          • RootVolume (boolean) --

            Specifies whether the configuration is for the cluster's root or secondary Amazon EBS volume. You can specify two ClusterEbsVolumeConfig fields to configure both the root and secondary volumes. Set the value to True if you'd like to provide your own customer managed Amazon Web Services KMS key to encrypt the root volume. When True:

            • The configuration is applied to the root volume.

            • You can't specify the VolumeSizeInGB field. The size of the root volume is determined for you.

            • You must specify a KMS key ID for VolumeKmsKeyId to encrypt the root volume with your own KMS key instead of an Amazon Web Services owned KMS key.

            Otherwise, by default, the value is False, and the following applies:

            • The configuration is applied to the secondary volume, while the root volume is encrypted with an Amazon Web Services owned key.

            • You must specify the VolumeSizeInGB field.

            • You can optionally specify the VolumeKmsKeyId to encrypt the secondary volume with your own KMS key instead of an Amazon Web Services owned KMS key.

    • OnStartDeepHealthChecks (list) --

      A flag indicating whether deep health checks should be performed when the cluster instance group is created or updated.

      • (string) --

    • TrainingPlanArn (string) --

      The Amazon Resource Name (ARN); of the training plan to use for this cluster instance group.

      For more information about how to reserve GPU capacity for your SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see ``CreateTrainingPlan ``.

    • OverrideVpcConfig (dict) --

      To configure multi-AZ deployments, customize the Amazon VPC configuration at the instance group level. You can specify different subnets and security groups across different AZs in the instance group specification to override a SageMaker HyperPod cluster's default Amazon VPC configuration. For more information about deploying a cluster in multiple AZs, see Setting up SageMaker HyperPod clusters across multiple AZs.

      • SecurityGroupIds (list) -- [REQUIRED]

        The VPC security group IDs, in the form sg-xxxxxxxx. Specify the security groups for the VPC that is specified in the Subnets field.

        • (string) --

      • Subnets (list) -- [REQUIRED]

        The ID of the subnets in the VPC to which you want to connect your training job or model. For information about the availability of specific instance types, see Supported Instance Types and Availability Zones.

        • (string) --

    • ScheduledUpdateConfig (dict) --

      The configuration object of the schedule that SageMaker uses to update the AMI.

      • ScheduleExpression (string) -- [REQUIRED]

        A cron expression that specifies the schedule that SageMaker follows when updating the AMI.

      • DeploymentConfig (dict) --

        The configuration to use when updating the AMI versions.

        • RollingUpdatePolicy (dict) --

          The policy that SageMaker uses when updating the AMI versions of the cluster.

          • MaximumBatchSize (dict) -- [REQUIRED]

            The maximum amount of instances in the cluster that SageMaker can update at a time.

            • Type (string) -- [REQUIRED]

              Specifies whether SageMaker should process the update by amount or percentage of instances.

            • Value (integer) -- [REQUIRED]

              Specifies the amount or percentage of instances SageMaker updates at a time.

          • RollbackMaximumBatchSize (dict) --

            The maximum amount of instances in the cluster that SageMaker can roll back at a time.

            • Type (string) -- [REQUIRED]

              Specifies whether SageMaker should process the update by amount or percentage of instances.

            • Value (integer) -- [REQUIRED]

              Specifies the amount or percentage of instances SageMaker updates at a time.

        • WaitIntervalInSeconds (integer) --

          The duration in seconds that SageMaker waits before updating more instances in the cluster.

        • AutoRollbackConfiguration (list) --

          An array that contains the alarms that SageMaker monitors to know whether to roll back the AMI update.

          • (dict) --

            The details of the alarm to monitor during the AMI update.

            • AlarmName (string) -- [REQUIRED]

              The name of the alarm.

    • ImageId (string) --

      When configuring your HyperPod cluster, you can specify an image ID using one of the following options:

      • HyperPodPublicAmiId: Use a HyperPod public AMI

      • CustomAmiId: Use your custom AMI

      • default: Use the default latest system image

      If you choose to use a custom AMI ( CustomAmiId), ensure it meets the following requirements:

      • Encryption: The custom AMI must be unencrypted.

      • Ownership: The custom AMI must be owned by the same Amazon Web Services account that is creating the HyperPod cluster.

      • Volume support: Only the primary AMI snapshot volume is supported; additional AMI volumes are not supported.

      When updating the instance group's AMI through the UpdateClusterSoftware operation, if an instance group uses a custom AMI, you must provide an ImageId or use the default as input. Note that if you don't specify an instance group in your UpdateClusterSoftware request, then all of the instance groups are patched with the specified image.

type RestrictedInstanceGroups:

list

param RestrictedInstanceGroups:

The specialized instance groups for training models like Amazon Nova to be created in the SageMaker HyperPod cluster.

  • (dict) --

    The specifications of a restricted instance group that you need to define.

    • InstanceCount (integer) -- [REQUIRED]

      Specifies the number of instances to add to the restricted instance group of a SageMaker HyperPod cluster.

    • InstanceGroupName (string) -- [REQUIRED]

      Specifies the name of the restricted instance group.

    • InstanceType (string) -- [REQUIRED]

      Specifies the instance type of the restricted instance group.

    • ExecutionRole (string) -- [REQUIRED]

      Specifies an IAM execution role to be assumed by the restricted instance group.

    • ThreadsPerCore (integer) --

      The number you specified to TreadsPerCore in CreateCluster for enabling or disabling multithreading. For instance types that support multithreading, you can specify 1 for disabling multithreading and 2 for enabling multithreading. For more information, see the reference table of CPU cores and threads per CPU core per instance type in the Amazon Elastic Compute Cloud User Guide.

    • InstanceStorageConfigs (list) --

      Specifies the additional storage configurations for the instances in the SageMaker HyperPod cluster restricted instance group.

      • (dict) --

        Defines the configuration for attaching additional storage to the instances in the SageMaker HyperPod cluster instance group. To learn more, see SageMaker HyperPod release notes: June 20, 2024.

        • EbsVolumeConfig (dict) --

          Defines the configuration for attaching additional Amazon Elastic Block Store (EBS) volumes to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to /opt/sagemaker.

          • VolumeSizeInGB (integer) --

            The size in gigabytes (GB) of the additional EBS volume to be attached to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to /opt/sagemaker.

          • VolumeKmsKeyId (string) --

            The ID of a KMS key to encrypt the Amazon EBS volume.

          • RootVolume (boolean) --

            Specifies whether the configuration is for the cluster's root or secondary Amazon EBS volume. You can specify two ClusterEbsVolumeConfig fields to configure both the root and secondary volumes. Set the value to True if you'd like to provide your own customer managed Amazon Web Services KMS key to encrypt the root volume. When True:

            • The configuration is applied to the root volume.

            • You can't specify the VolumeSizeInGB field. The size of the root volume is determined for you.

            • You must specify a KMS key ID for VolumeKmsKeyId to encrypt the root volume with your own KMS key instead of an Amazon Web Services owned KMS key.

            Otherwise, by default, the value is False, and the following applies:

            • The configuration is applied to the secondary volume, while the root volume is encrypted with an Amazon Web Services owned key.

            • You must specify the VolumeSizeInGB field.

            • You can optionally specify the VolumeKmsKeyId to encrypt the secondary volume with your own KMS key instead of an Amazon Web Services owned KMS key.

    • OnStartDeepHealthChecks (list) --

      A flag indicating whether deep health checks should be performed when the cluster restricted instance group is created or updated.

      • (string) --

    • TrainingPlanArn (string) --

      The Amazon Resource Name (ARN) of the training plan to filter clusters by. For more information about reserving GPU capacity for your SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see ``CreateTrainingPlan ``.

    • OverrideVpcConfig (dict) --

      Specifies an Amazon Virtual Private Cloud (VPC) that your SageMaker jobs, hosted models, and compute resources have access to. You can control access to and from your resources by configuring a VPC. For more information, see Give SageMaker Access to Resources in your Amazon VPC.

      • SecurityGroupIds (list) -- [REQUIRED]

        The VPC security group IDs, in the form sg-xxxxxxxx. Specify the security groups for the VPC that is specified in the Subnets field.

        • (string) --

      • Subnets (list) -- [REQUIRED]

        The ID of the subnets in the VPC to which you want to connect your training job or model. For information about the availability of specific instance types, see Supported Instance Types and Availability Zones.

        • (string) --

    • ScheduledUpdateConfig (dict) --

      The configuration object of the schedule that SageMaker follows when updating the AMI.

      • ScheduleExpression (string) -- [REQUIRED]

        A cron expression that specifies the schedule that SageMaker follows when updating the AMI.

      • DeploymentConfig (dict) --

        The configuration to use when updating the AMI versions.

        • RollingUpdatePolicy (dict) --

          The policy that SageMaker uses when updating the AMI versions of the cluster.

          • MaximumBatchSize (dict) -- [REQUIRED]

            The maximum amount of instances in the cluster that SageMaker can update at a time.

            • Type (string) -- [REQUIRED]

              Specifies whether SageMaker should process the update by amount or percentage of instances.

            • Value (integer) -- [REQUIRED]

              Specifies the amount or percentage of instances SageMaker updates at a time.

          • RollbackMaximumBatchSize (dict) --

            The maximum amount of instances in the cluster that SageMaker can roll back at a time.

            • Type (string) -- [REQUIRED]

              Specifies whether SageMaker should process the update by amount or percentage of instances.

            • Value (integer) -- [REQUIRED]

              Specifies the amount or percentage of instances SageMaker updates at a time.

        • WaitIntervalInSeconds (integer) --

          The duration in seconds that SageMaker waits before updating more instances in the cluster.

        • AutoRollbackConfiguration (list) --

          An array that contains the alarms that SageMaker monitors to know whether to roll back the AMI update.

          • (dict) --

            The details of the alarm to monitor during the AMI update.

            • AlarmName (string) -- [REQUIRED]

              The name of the alarm.

    • EnvironmentConfig (dict) -- [REQUIRED]

      The configuration for the restricted instance groups (RIG) environment.

      • FSxLustreConfig (dict) --

        Configuration settings for an Amazon FSx for Lustre file system to be used with the cluster.

        • SizeInGiB (integer) -- [REQUIRED]

          The storage capacity of the Amazon FSx for Lustre file system, specified in gibibytes (GiB).

        • PerUnitStorageThroughput (integer) -- [REQUIRED]

          The throughput capacity of the Amazon FSx for Lustre file system, measured in MB/s per TiB of storage.

type VpcConfig:

dict

param VpcConfig:

Specifies the Amazon Virtual Private Cloud (VPC) that is associated with the Amazon SageMaker HyperPod cluster. You can control access to and from your resources by configuring your VPC. For more information, see Give SageMaker access to resources in your Amazon VPC.

  • SecurityGroupIds (list) -- [REQUIRED]

    The VPC security group IDs, in the form sg-xxxxxxxx. Specify the security groups for the VPC that is specified in the Subnets field.

    • (string) --

  • Subnets (list) -- [REQUIRED]

    The ID of the subnets in the VPC to which you want to connect your training job or model. For information about the availability of specific instance types, see Supported Instance Types and Availability Zones.

    • (string) --

type Tags:

list

param Tags:

Custom tags for managing the SageMaker HyperPod cluster as an Amazon Web Services resource. You can add tags to your cluster in the same way you add them in other Amazon Web Services services that support tagging. To learn more about tagging Amazon Web Services resources in general, see Tagging Amazon Web Services Resources User Guide.

  • (dict) --

    A tag object that consists of a key and an optional value, used to manage metadata for SageMaker Amazon Web Services resources.

    You can add tags to notebook instances, training jobs, hyperparameter tuning jobs, batch transform jobs, models, labeling jobs, work teams, endpoint configurations, and endpoints. For more information on adding tags to SageMaker resources, see AddTags.

    For more information on adding metadata to your Amazon Web Services resources with tagging, see Tagging Amazon Web Services resources. For advice on best practices for managing Amazon Web Services resources with tagging, see Tagging Best Practices: Implement an Effective Amazon Web Services Resource Tagging Strategy.

    • Key (string) -- [REQUIRED]

      The tag key. Tag keys must be unique per resource.

    • Value (string) -- [REQUIRED]

      The tag value.

type Orchestrator:

dict

param Orchestrator:

The type of orchestrator to use for the SageMaker HyperPod cluster. Currently, the only supported value is "eks", which is to use an Amazon Elastic Kubernetes Service cluster as the orchestrator.

  • Eks (dict) -- [REQUIRED]

    The Amazon EKS cluster used as the orchestrator for the SageMaker HyperPod cluster.

    • ClusterArn (string) -- [REQUIRED]

      The Amazon Resource Name (ARN) of the Amazon EKS cluster associated with the SageMaker HyperPod cluster.

type NodeRecovery:

string

param NodeRecovery:

The node recovery mode for the SageMaker HyperPod cluster. When set to Automatic, SageMaker HyperPod will automatically reboot or replace faulty nodes when issues are detected. When set to None, cluster administrators will need to manually manage any faulty cluster instances.

type NodeProvisioningMode:

string

param NodeProvisioningMode:

The mode for provisioning nodes in the cluster. You can specify the following modes:

  • Continuous: Scaling behavior that enables 1) concurrent operation execution within instance groups, 2) continuous retry mechanisms for failed operations, 3) enhanced customer visibility into cluster events through detailed event streams, 4) partial provisioning capabilities. Your clusters and instance groups remain InService while scaling. This mode is only supported for EKS orchestrated clusters.

type ClusterRole:

string

param ClusterRole:

The Amazon Resource Name (ARN) of the IAM role that HyperPod assumes to perform cluster autoscaling operations. This role must have permissions for sagemaker:BatchAddClusterNodes and sagemaker:BatchDeleteClusterNodes. This is only required when autoscaling is enabled and when HyperPod is performing autoscaling operations.

type AutoScaling:

dict

param AutoScaling:

The autoscaling configuration for the cluster. Enables automatic scaling of cluster nodes based on workload demand using a Karpenter-based system.

  • Mode (string) -- [REQUIRED]

    Describes whether autoscaling is enabled or disabled for the cluster. Valid values are Enable and Disable.

  • AutoScalerType (string) --

    The type of autoscaler to use. Currently supported value is Karpenter.

rtype:

dict

returns:

Response Syntax

{
    'ClusterArn': 'string'
}

Response Structure

  • (dict) --

    • ClusterArn (string) --

      The Amazon Resource Name (ARN) of the cluster.

DescribeCluster (updated) Link ¶
Changes (response)
{'AutoScaling': {'AutoScalerType': 'Karpenter',
                 'FailureMessage': 'string',
                 'Mode': 'Enable | Disable',
                 'Status': 'InService | Failed | Creating | Deleting'},
 'ClusterRole': 'string'}

Retrieves information of a SageMaker HyperPod cluster.

See also: AWS API Documentation

Request Syntax

client.describe_cluster(
    ClusterName='string'
)
type ClusterName:

string

param ClusterName:

[REQUIRED]

The string name or the Amazon Resource Name (ARN) of the SageMaker HyperPod cluster.

rtype:

dict

returns:

Response Syntax

{
    'ClusterArn': 'string',
    'ClusterName': 'string',
    'ClusterStatus': 'Creating'|'Deleting'|'Failed'|'InService'|'RollingBack'|'SystemUpdating'|'Updating',
    'CreationTime': datetime(2015, 1, 1),
    'FailureMessage': 'string',
    'InstanceGroups': [
        {
            'CurrentCount': 123,
            'TargetCount': 123,
            'InstanceGroupName': 'string',
            'InstanceType': 'ml.p4d.24xlarge'|'ml.p4de.24xlarge'|'ml.p5.48xlarge'|'ml.p6e-gb200.36xlarge'|'ml.trn1.32xlarge'|'ml.trn1n.32xlarge'|'ml.g5.xlarge'|'ml.g5.2xlarge'|'ml.g5.4xlarge'|'ml.g5.8xlarge'|'ml.g5.12xlarge'|'ml.g5.16xlarge'|'ml.g5.24xlarge'|'ml.g5.48xlarge'|'ml.c5.large'|'ml.c5.xlarge'|'ml.c5.2xlarge'|'ml.c5.4xlarge'|'ml.c5.9xlarge'|'ml.c5.12xlarge'|'ml.c5.18xlarge'|'ml.c5.24xlarge'|'ml.c5n.large'|'ml.c5n.2xlarge'|'ml.c5n.4xlarge'|'ml.c5n.9xlarge'|'ml.c5n.18xlarge'|'ml.m5.large'|'ml.m5.xlarge'|'ml.m5.2xlarge'|'ml.m5.4xlarge'|'ml.m5.8xlarge'|'ml.m5.12xlarge'|'ml.m5.16xlarge'|'ml.m5.24xlarge'|'ml.t3.medium'|'ml.t3.large'|'ml.t3.xlarge'|'ml.t3.2xlarge'|'ml.g6.xlarge'|'ml.g6.2xlarge'|'ml.g6.4xlarge'|'ml.g6.8xlarge'|'ml.g6.16xlarge'|'ml.g6.12xlarge'|'ml.g6.24xlarge'|'ml.g6.48xlarge'|'ml.gr6.4xlarge'|'ml.gr6.8xlarge'|'ml.g6e.xlarge'|'ml.g6e.2xlarge'|'ml.g6e.4xlarge'|'ml.g6e.8xlarge'|'ml.g6e.16xlarge'|'ml.g6e.12xlarge'|'ml.g6e.24xlarge'|'ml.g6e.48xlarge'|'ml.p5e.48xlarge'|'ml.p5en.48xlarge'|'ml.p6-b200.48xlarge'|'ml.trn2.48xlarge'|'ml.c6i.large'|'ml.c6i.xlarge'|'ml.c6i.2xlarge'|'ml.c6i.4xlarge'|'ml.c6i.8xlarge'|'ml.c6i.12xlarge'|'ml.c6i.16xlarge'|'ml.c6i.24xlarge'|'ml.c6i.32xlarge'|'ml.m6i.large'|'ml.m6i.xlarge'|'ml.m6i.2xlarge'|'ml.m6i.4xlarge'|'ml.m6i.8xlarge'|'ml.m6i.12xlarge'|'ml.m6i.16xlarge'|'ml.m6i.24xlarge'|'ml.m6i.32xlarge'|'ml.r6i.large'|'ml.r6i.xlarge'|'ml.r6i.2xlarge'|'ml.r6i.4xlarge'|'ml.r6i.8xlarge'|'ml.r6i.12xlarge'|'ml.r6i.16xlarge'|'ml.r6i.24xlarge'|'ml.r6i.32xlarge'|'ml.i3en.large'|'ml.i3en.xlarge'|'ml.i3en.2xlarge'|'ml.i3en.3xlarge'|'ml.i3en.6xlarge'|'ml.i3en.12xlarge'|'ml.i3en.24xlarge'|'ml.m7i.large'|'ml.m7i.xlarge'|'ml.m7i.2xlarge'|'ml.m7i.4xlarge'|'ml.m7i.8xlarge'|'ml.m7i.12xlarge'|'ml.m7i.16xlarge'|'ml.m7i.24xlarge'|'ml.m7i.48xlarge'|'ml.r7i.large'|'ml.r7i.xlarge'|'ml.r7i.2xlarge'|'ml.r7i.4xlarge'|'ml.r7i.8xlarge'|'ml.r7i.12xlarge'|'ml.r7i.16xlarge'|'ml.r7i.24xlarge'|'ml.r7i.48xlarge',
            'LifeCycleConfig': {
                'SourceS3Uri': 'string',
                'OnCreate': 'string'
            },
            'ExecutionRole': 'string',
            'ThreadsPerCore': 123,
            'InstanceStorageConfigs': [
                {
                    'EbsVolumeConfig': {
                        'VolumeSizeInGB': 123,
                        'VolumeKmsKeyId': 'string',
                        'RootVolume': True|False
                    }
                },
            ],
            'OnStartDeepHealthChecks': [
                'InstanceStress'|'InstanceConnectivity',
            ],
            'Status': 'InService'|'Creating'|'Updating'|'Failed'|'Degraded'|'SystemUpdating'|'Deleting',
            'TrainingPlanArn': 'string',
            'TrainingPlanStatus': 'string',
            'OverrideVpcConfig': {
                'SecurityGroupIds': [
                    'string',
                ],
                'Subnets': [
                    'string',
                ]
            },
            'ScheduledUpdateConfig': {
                'ScheduleExpression': 'string',
                'DeploymentConfig': {
                    'RollingUpdatePolicy': {
                        'MaximumBatchSize': {
                            'Type': 'INSTANCE_COUNT'|'CAPACITY_PERCENTAGE',
                            'Value': 123
                        },
                        'RollbackMaximumBatchSize': {
                            'Type': 'INSTANCE_COUNT'|'CAPACITY_PERCENTAGE',
                            'Value': 123
                        }
                    },
                    'WaitIntervalInSeconds': 123,
                    'AutoRollbackConfiguration': [
                        {
                            'AlarmName': 'string'
                        },
                    ]
                }
            },
            'CurrentImageId': 'string',
            'DesiredImageId': 'string'
        },
    ],
    'RestrictedInstanceGroups': [
        {
            'CurrentCount': 123,
            'TargetCount': 123,
            'InstanceGroupName': 'string',
            'InstanceType': 'ml.p4d.24xlarge'|'ml.p4de.24xlarge'|'ml.p5.48xlarge'|'ml.p6e-gb200.36xlarge'|'ml.trn1.32xlarge'|'ml.trn1n.32xlarge'|'ml.g5.xlarge'|'ml.g5.2xlarge'|'ml.g5.4xlarge'|'ml.g5.8xlarge'|'ml.g5.12xlarge'|'ml.g5.16xlarge'|'ml.g5.24xlarge'|'ml.g5.48xlarge'|'ml.c5.large'|'ml.c5.xlarge'|'ml.c5.2xlarge'|'ml.c5.4xlarge'|'ml.c5.9xlarge'|'ml.c5.12xlarge'|'ml.c5.18xlarge'|'ml.c5.24xlarge'|'ml.c5n.large'|'ml.c5n.2xlarge'|'ml.c5n.4xlarge'|'ml.c5n.9xlarge'|'ml.c5n.18xlarge'|'ml.m5.large'|'ml.m5.xlarge'|'ml.m5.2xlarge'|'ml.m5.4xlarge'|'ml.m5.8xlarge'|'ml.m5.12xlarge'|'ml.m5.16xlarge'|'ml.m5.24xlarge'|'ml.t3.medium'|'ml.t3.large'|'ml.t3.xlarge'|'ml.t3.2xlarge'|'ml.g6.xlarge'|'ml.g6.2xlarge'|'ml.g6.4xlarge'|'ml.g6.8xlarge'|'ml.g6.16xlarge'|'ml.g6.12xlarge'|'ml.g6.24xlarge'|'ml.g6.48xlarge'|'ml.gr6.4xlarge'|'ml.gr6.8xlarge'|'ml.g6e.xlarge'|'ml.g6e.2xlarge'|'ml.g6e.4xlarge'|'ml.g6e.8xlarge'|'ml.g6e.16xlarge'|'ml.g6e.12xlarge'|'ml.g6e.24xlarge'|'ml.g6e.48xlarge'|'ml.p5e.48xlarge'|'ml.p5en.48xlarge'|'ml.p6-b200.48xlarge'|'ml.trn2.48xlarge'|'ml.c6i.large'|'ml.c6i.xlarge'|'ml.c6i.2xlarge'|'ml.c6i.4xlarge'|'ml.c6i.8xlarge'|'ml.c6i.12xlarge'|'ml.c6i.16xlarge'|'ml.c6i.24xlarge'|'ml.c6i.32xlarge'|'ml.m6i.large'|'ml.m6i.xlarge'|'ml.m6i.2xlarge'|'ml.m6i.4xlarge'|'ml.m6i.8xlarge'|'ml.m6i.12xlarge'|'ml.m6i.16xlarge'|'ml.m6i.24xlarge'|'ml.m6i.32xlarge'|'ml.r6i.large'|'ml.r6i.xlarge'|'ml.r6i.2xlarge'|'ml.r6i.4xlarge'|'ml.r6i.8xlarge'|'ml.r6i.12xlarge'|'ml.r6i.16xlarge'|'ml.r6i.24xlarge'|'ml.r6i.32xlarge'|'ml.i3en.large'|'ml.i3en.xlarge'|'ml.i3en.2xlarge'|'ml.i3en.3xlarge'|'ml.i3en.6xlarge'|'ml.i3en.12xlarge'|'ml.i3en.24xlarge'|'ml.m7i.large'|'ml.m7i.xlarge'|'ml.m7i.2xlarge'|'ml.m7i.4xlarge'|'ml.m7i.8xlarge'|'ml.m7i.12xlarge'|'ml.m7i.16xlarge'|'ml.m7i.24xlarge'|'ml.m7i.48xlarge'|'ml.r7i.large'|'ml.r7i.xlarge'|'ml.r7i.2xlarge'|'ml.r7i.4xlarge'|'ml.r7i.8xlarge'|'ml.r7i.12xlarge'|'ml.r7i.16xlarge'|'ml.r7i.24xlarge'|'ml.r7i.48xlarge',
            'ExecutionRole': 'string',
            'ThreadsPerCore': 123,
            'InstanceStorageConfigs': [
                {
                    'EbsVolumeConfig': {
                        'VolumeSizeInGB': 123,
                        'VolumeKmsKeyId': 'string',
                        'RootVolume': True|False
                    }
                },
            ],
            'OnStartDeepHealthChecks': [
                'InstanceStress'|'InstanceConnectivity',
            ],
            'Status': 'InService'|'Creating'|'Updating'|'Failed'|'Degraded'|'SystemUpdating'|'Deleting',
            'TrainingPlanArn': 'string',
            'TrainingPlanStatus': 'string',
            'OverrideVpcConfig': {
                'SecurityGroupIds': [
                    'string',
                ],
                'Subnets': [
                    'string',
                ]
            },
            'ScheduledUpdateConfig': {
                'ScheduleExpression': 'string',
                'DeploymentConfig': {
                    'RollingUpdatePolicy': {
                        'MaximumBatchSize': {
                            'Type': 'INSTANCE_COUNT'|'CAPACITY_PERCENTAGE',
                            'Value': 123
                        },
                        'RollbackMaximumBatchSize': {
                            'Type': 'INSTANCE_COUNT'|'CAPACITY_PERCENTAGE',
                            'Value': 123
                        }
                    },
                    'WaitIntervalInSeconds': 123,
                    'AutoRollbackConfiguration': [
                        {
                            'AlarmName': 'string'
                        },
                    ]
                }
            },
            'EnvironmentConfig': {
                'FSxLustreConfig': {
                    'SizeInGiB': 123,
                    'PerUnitStorageThroughput': 123
                },
                'S3OutputPath': 'string'
            }
        },
    ],
    'VpcConfig': {
        'SecurityGroupIds': [
            'string',
        ],
        'Subnets': [
            'string',
        ]
    },
    'Orchestrator': {
        'Eks': {
            'ClusterArn': 'string'
        }
    },
    'NodeRecovery': 'Automatic'|'None',
    'NodeProvisioningMode': 'Continuous',
    'ClusterRole': 'string',
    'AutoScaling': {
        'Mode': 'Enable'|'Disable',
        'AutoScalerType': 'Karpenter',
        'Status': 'InService'|'Failed'|'Creating'|'Deleting',
        'FailureMessage': 'string'
    }
}

Response Structure

  • (dict) --

    • ClusterArn (string) --

      The Amazon Resource Name (ARN) of the SageMaker HyperPod cluster.

    • ClusterName (string) --

      The name of the SageMaker HyperPod cluster.

    • ClusterStatus (string) --

      The status of the SageMaker HyperPod cluster.

    • CreationTime (datetime) --

      The time when the SageMaker Cluster is created.

    • FailureMessage (string) --

      The failure message of the SageMaker HyperPod cluster.

    • InstanceGroups (list) --

      The instance groups of the SageMaker HyperPod cluster.

      • (dict) --

        Details of an instance group in a SageMaker HyperPod cluster.

        • CurrentCount (integer) --

          The number of instances that are currently in the instance group of a SageMaker HyperPod cluster.

        • TargetCount (integer) --

          The number of instances you specified to add to the instance group of a SageMaker HyperPod cluster.

        • InstanceGroupName (string) --

          The name of the instance group of a SageMaker HyperPod cluster.

        • InstanceType (string) --

          The instance type of the instance group of a SageMaker HyperPod cluster.

        • LifeCycleConfig (dict) --

          Details of LifeCycle configuration for the instance group.

          • SourceS3Uri (string) --

            An Amazon S3 bucket path where your lifecycle scripts are stored.

          • OnCreate (string) --

            The file name of the entrypoint script of lifecycle scripts under SourceS3Uri. This entrypoint script runs during cluster creation.

        • ExecutionRole (string) --

          The execution role for the instance group to assume.

        • ThreadsPerCore (integer) --

          The number you specified to TreadsPerCore in CreateCluster for enabling or disabling multithreading. For instance types that support multithreading, you can specify 1 for disabling multithreading and 2 for enabling multithreading. For more information, see the reference table of CPU cores and threads per CPU core per instance type in the Amazon Elastic Compute Cloud User Guide.

        • InstanceStorageConfigs (list) --

          The additional storage configurations for the instances in the SageMaker HyperPod cluster instance group.

          • (dict) --

            Defines the configuration for attaching additional storage to the instances in the SageMaker HyperPod cluster instance group. To learn more, see SageMaker HyperPod release notes: June 20, 2024.

            • EbsVolumeConfig (dict) --

              Defines the configuration for attaching additional Amazon Elastic Block Store (EBS) volumes to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to /opt/sagemaker.

              • VolumeSizeInGB (integer) --

                The size in gigabytes (GB) of the additional EBS volume to be attached to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to /opt/sagemaker.

              • VolumeKmsKeyId (string) --

                The ID of a KMS key to encrypt the Amazon EBS volume.

              • RootVolume (boolean) --

                Specifies whether the configuration is for the cluster's root or secondary Amazon EBS volume. You can specify two ClusterEbsVolumeConfig fields to configure both the root and secondary volumes. Set the value to True if you'd like to provide your own customer managed Amazon Web Services KMS key to encrypt the root volume. When True:

                • The configuration is applied to the root volume.

                • You can't specify the VolumeSizeInGB field. The size of the root volume is determined for you.

                • You must specify a KMS key ID for VolumeKmsKeyId to encrypt the root volume with your own KMS key instead of an Amazon Web Services owned KMS key.

                Otherwise, by default, the value is False, and the following applies:

                • The configuration is applied to the secondary volume, while the root volume is encrypted with an Amazon Web Services owned key.

                • You must specify the VolumeSizeInGB field.

                • You can optionally specify the VolumeKmsKeyId to encrypt the secondary volume with your own KMS key instead of an Amazon Web Services owned KMS key.

        • OnStartDeepHealthChecks (list) --

          A flag indicating whether deep health checks should be performed when the cluster instance group is created or updated.

          • (string) --

        • Status (string) --

          The current status of the cluster instance group.

          • InService: The instance group is active and healthy.

          • Creating: The instance group is being provisioned.

          • Updating: The instance group is being updated.

          • Failed: The instance group has failed to provision or is no longer healthy.

          • Degraded: The instance group is degraded, meaning that some instances have failed to provision or are no longer healthy.

          • Deleting: The instance group is being deleted.

        • TrainingPlanArn (string) --

          The Amazon Resource Name (ARN); of the training plan associated with this cluster instance group.

          For more information about how to reserve GPU capacity for your SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see ``CreateTrainingPlan ``.

        • TrainingPlanStatus (string) --

          The current status of the training plan associated with this cluster instance group.

        • OverrideVpcConfig (dict) --

          The customized Amazon VPC configuration at the instance group level that overrides the default Amazon VPC configuration of the SageMaker HyperPod cluster.

          • SecurityGroupIds (list) --

            The VPC security group IDs, in the form sg-xxxxxxxx. Specify the security groups for the VPC that is specified in the Subnets field.

            • (string) --

          • Subnets (list) --

            The ID of the subnets in the VPC to which you want to connect your training job or model. For information about the availability of specific instance types, see Supported Instance Types and Availability Zones.

            • (string) --

        • ScheduledUpdateConfig (dict) --

          The configuration object of the schedule that SageMaker follows when updating the AMI.

          • ScheduleExpression (string) --

            A cron expression that specifies the schedule that SageMaker follows when updating the AMI.

          • DeploymentConfig (dict) --

            The configuration to use when updating the AMI versions.

            • RollingUpdatePolicy (dict) --

              The policy that SageMaker uses when updating the AMI versions of the cluster.

              • MaximumBatchSize (dict) --

                The maximum amount of instances in the cluster that SageMaker can update at a time.

                • Type (string) --

                  Specifies whether SageMaker should process the update by amount or percentage of instances.

                • Value (integer) --

                  Specifies the amount or percentage of instances SageMaker updates at a time.

              • RollbackMaximumBatchSize (dict) --

                The maximum amount of instances in the cluster that SageMaker can roll back at a time.

                • Type (string) --

                  Specifies whether SageMaker should process the update by amount or percentage of instances.

                • Value (integer) --

                  Specifies the amount or percentage of instances SageMaker updates at a time.

            • WaitIntervalInSeconds (integer) --

              The duration in seconds that SageMaker waits before updating more instances in the cluster.

            • AutoRollbackConfiguration (list) --

              An array that contains the alarms that SageMaker monitors to know whether to roll back the AMI update.

              • (dict) --

                The details of the alarm to monitor during the AMI update.

                • AlarmName (string) --

                  The name of the alarm.

        • CurrentImageId (string) --

          The ID of the Amazon Machine Image (AMI) currently in use by the instance group.

        • DesiredImageId (string) --

          The ID of the Amazon Machine Image (AMI) desired for the instance group.

    • RestrictedInstanceGroups (list) --

      The specialized instance groups for training models like Amazon Nova to be created in the SageMaker HyperPod cluster.

      • (dict) --

        The instance group details of the restricted instance group (RIG).

        • CurrentCount (integer) --

          The number of instances that are currently in the restricted instance group of a SageMaker HyperPod cluster.

        • TargetCount (integer) --

          The number of instances you specified to add to the restricted instance group of a SageMaker HyperPod cluster.

        • InstanceGroupName (string) --

          The name of the restricted instance group of a SageMaker HyperPod cluster.

        • InstanceType (string) --

          The instance type of the restricted instance group of a SageMaker HyperPod cluster.

        • ExecutionRole (string) --

          The execution role for the restricted instance group to assume.

        • ThreadsPerCore (integer) --

          The number you specified to TreadsPerCore in CreateCluster for enabling or disabling multithreading. For instance types that support multithreading, you can specify 1 for disabling multithreading and 2 for enabling multithreading. For more information, see the reference table of CPU cores and threads per CPU core per instance type in the Amazon Elastic Compute Cloud User Guide.

        • InstanceStorageConfigs (list) --

          The additional storage configurations for the instances in the SageMaker HyperPod cluster restricted instance group.

          • (dict) --

            Defines the configuration for attaching additional storage to the instances in the SageMaker HyperPod cluster instance group. To learn more, see SageMaker HyperPod release notes: June 20, 2024.

            • EbsVolumeConfig (dict) --

              Defines the configuration for attaching additional Amazon Elastic Block Store (EBS) volumes to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to /opt/sagemaker.

              • VolumeSizeInGB (integer) --

                The size in gigabytes (GB) of the additional EBS volume to be attached to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to /opt/sagemaker.

              • VolumeKmsKeyId (string) --

                The ID of a KMS key to encrypt the Amazon EBS volume.

              • RootVolume (boolean) --

                Specifies whether the configuration is for the cluster's root or secondary Amazon EBS volume. You can specify two ClusterEbsVolumeConfig fields to configure both the root and secondary volumes. Set the value to True if you'd like to provide your own customer managed Amazon Web Services KMS key to encrypt the root volume. When True:

                • The configuration is applied to the root volume.

                • You can't specify the VolumeSizeInGB field. The size of the root volume is determined for you.

                • You must specify a KMS key ID for VolumeKmsKeyId to encrypt the root volume with your own KMS key instead of an Amazon Web Services owned KMS key.

                Otherwise, by default, the value is False, and the following applies:

                • The configuration is applied to the secondary volume, while the root volume is encrypted with an Amazon Web Services owned key.

                • You must specify the VolumeSizeInGB field.

                • You can optionally specify the VolumeKmsKeyId to encrypt the secondary volume with your own KMS key instead of an Amazon Web Services owned KMS key.

        • OnStartDeepHealthChecks (list) --

          A flag indicating whether deep health checks should be performed when the cluster's restricted instance group is created or updated.

          • (string) --

        • Status (string) --

          The current status of the cluster's restricted instance group.

          • InService: The restricted instance group is active and healthy.

          • Creating: The restricted instance group is being provisioned.

          • Updating: The restricted instance group is being updated.

          • Failed: The restricted instance group has failed to provision or is no longer healthy.

          • Degraded: The restricted instance group is degraded, meaning that some instances have failed to provision or are no longer healthy.

          • Deleting: The restricted instance group is being deleted.

        • TrainingPlanArn (string) --

          The Amazon Resource Name (ARN) of the training plan to filter clusters by. For more information about reserving GPU capacity for your SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see ``CreateTrainingPlan ``.

        • TrainingPlanStatus (string) --

          The current status of the training plan associated with this cluster restricted instance group.

        • OverrideVpcConfig (dict) --

          Specifies an Amazon Virtual Private Cloud (VPC) that your SageMaker jobs, hosted models, and compute resources have access to. You can control access to and from your resources by configuring a VPC. For more information, see Give SageMaker Access to Resources in your Amazon VPC.

          • SecurityGroupIds (list) --

            The VPC security group IDs, in the form sg-xxxxxxxx. Specify the security groups for the VPC that is specified in the Subnets field.

            • (string) --

          • Subnets (list) --

            The ID of the subnets in the VPC to which you want to connect your training job or model. For information about the availability of specific instance types, see Supported Instance Types and Availability Zones.

            • (string) --

        • ScheduledUpdateConfig (dict) --

          The configuration object of the schedule that SageMaker follows when updating the AMI.

          • ScheduleExpression (string) --

            A cron expression that specifies the schedule that SageMaker follows when updating the AMI.

          • DeploymentConfig (dict) --

            The configuration to use when updating the AMI versions.

            • RollingUpdatePolicy (dict) --

              The policy that SageMaker uses when updating the AMI versions of the cluster.

              • MaximumBatchSize (dict) --

                The maximum amount of instances in the cluster that SageMaker can update at a time.

                • Type (string) --

                  Specifies whether SageMaker should process the update by amount or percentage of instances.

                • Value (integer) --

                  Specifies the amount or percentage of instances SageMaker updates at a time.

              • RollbackMaximumBatchSize (dict) --

                The maximum amount of instances in the cluster that SageMaker can roll back at a time.

                • Type (string) --

                  Specifies whether SageMaker should process the update by amount or percentage of instances.

                • Value (integer) --

                  Specifies the amount or percentage of instances SageMaker updates at a time.

            • WaitIntervalInSeconds (integer) --

              The duration in seconds that SageMaker waits before updating more instances in the cluster.

            • AutoRollbackConfiguration (list) --

              An array that contains the alarms that SageMaker monitors to know whether to roll back the AMI update.

              • (dict) --

                The details of the alarm to monitor during the AMI update.

                • AlarmName (string) --

                  The name of the alarm.

        • EnvironmentConfig (dict) --

          The configuration for the restricted instance groups (RIG) environment.

          • FSxLustreConfig (dict) --

            Configuration settings for an Amazon FSx for Lustre file system to be used with the cluster.

            • SizeInGiB (integer) --

              The storage capacity of the Amazon FSx for Lustre file system, specified in gibibytes (GiB).

            • PerUnitStorageThroughput (integer) --

              The throughput capacity of the Amazon FSx for Lustre file system, measured in MB/s per TiB of storage.

          • S3OutputPath (string) --

            The Amazon S3 path where output data from the restricted instance group (RIG) environment will be stored.

    • VpcConfig (dict) --

      Specifies an Amazon Virtual Private Cloud (VPC) that your SageMaker jobs, hosted models, and compute resources have access to. You can control access to and from your resources by configuring a VPC. For more information, see Give SageMaker Access to Resources in your Amazon VPC.

      • SecurityGroupIds (list) --

        The VPC security group IDs, in the form sg-xxxxxxxx. Specify the security groups for the VPC that is specified in the Subnets field.

        • (string) --

      • Subnets (list) --

        The ID of the subnets in the VPC to which you want to connect your training job or model. For information about the availability of specific instance types, see Supported Instance Types and Availability Zones.

        • (string) --

    • Orchestrator (dict) --

      The type of orchestrator used for the SageMaker HyperPod cluster.

      • Eks (dict) --

        The Amazon EKS cluster used as the orchestrator for the SageMaker HyperPod cluster.

        • ClusterArn (string) --

          The Amazon Resource Name (ARN) of the Amazon EKS cluster associated with the SageMaker HyperPod cluster.

    • NodeRecovery (string) --

      The node recovery mode configured for the SageMaker HyperPod cluster.

    • NodeProvisioningMode (string) --

      The mode used for provisioning nodes in the cluster.

    • ClusterRole (string) --

      The Amazon Resource Name (ARN) of the IAM role that HyperPod uses for cluster autoscaling operations.

    • AutoScaling (dict) --

      The current autoscaling configuration and status for the autoscaler.

      • Mode (string) --

        Describes whether autoscaling is enabled or disabled for the cluster.

      • AutoScalerType (string) --

        The type of autoscaler configured for the cluster.

      • Status (string) --

        The current status of the autoscaling configuration. Valid values are InService, Failed, Creating, and Deleting.

      • FailureMessage (string) --

        If the autoscaling status is Failed, this field contains a message describing the failure.

UpdateCluster (updated) Link ¶
Changes (request)
{'AutoScaling': {'AutoScalerType': 'Karpenter', 'Mode': 'Enable | Disable'},
 'ClusterRole': 'string'}

Updates a SageMaker HyperPod cluster.

See also: AWS API Documentation

Request Syntax

client.update_cluster(
    ClusterName='string',
    InstanceGroups=[
        {
            'InstanceCount': 123,
            'InstanceGroupName': 'string',
            'InstanceType': 'ml.p4d.24xlarge'|'ml.p4de.24xlarge'|'ml.p5.48xlarge'|'ml.p6e-gb200.36xlarge'|'ml.trn1.32xlarge'|'ml.trn1n.32xlarge'|'ml.g5.xlarge'|'ml.g5.2xlarge'|'ml.g5.4xlarge'|'ml.g5.8xlarge'|'ml.g5.12xlarge'|'ml.g5.16xlarge'|'ml.g5.24xlarge'|'ml.g5.48xlarge'|'ml.c5.large'|'ml.c5.xlarge'|'ml.c5.2xlarge'|'ml.c5.4xlarge'|'ml.c5.9xlarge'|'ml.c5.12xlarge'|'ml.c5.18xlarge'|'ml.c5.24xlarge'|'ml.c5n.large'|'ml.c5n.2xlarge'|'ml.c5n.4xlarge'|'ml.c5n.9xlarge'|'ml.c5n.18xlarge'|'ml.m5.large'|'ml.m5.xlarge'|'ml.m5.2xlarge'|'ml.m5.4xlarge'|'ml.m5.8xlarge'|'ml.m5.12xlarge'|'ml.m5.16xlarge'|'ml.m5.24xlarge'|'ml.t3.medium'|'ml.t3.large'|'ml.t3.xlarge'|'ml.t3.2xlarge'|'ml.g6.xlarge'|'ml.g6.2xlarge'|'ml.g6.4xlarge'|'ml.g6.8xlarge'|'ml.g6.16xlarge'|'ml.g6.12xlarge'|'ml.g6.24xlarge'|'ml.g6.48xlarge'|'ml.gr6.4xlarge'|'ml.gr6.8xlarge'|'ml.g6e.xlarge'|'ml.g6e.2xlarge'|'ml.g6e.4xlarge'|'ml.g6e.8xlarge'|'ml.g6e.16xlarge'|'ml.g6e.12xlarge'|'ml.g6e.24xlarge'|'ml.g6e.48xlarge'|'ml.p5e.48xlarge'|'ml.p5en.48xlarge'|'ml.p6-b200.48xlarge'|'ml.trn2.48xlarge'|'ml.c6i.large'|'ml.c6i.xlarge'|'ml.c6i.2xlarge'|'ml.c6i.4xlarge'|'ml.c6i.8xlarge'|'ml.c6i.12xlarge'|'ml.c6i.16xlarge'|'ml.c6i.24xlarge'|'ml.c6i.32xlarge'|'ml.m6i.large'|'ml.m6i.xlarge'|'ml.m6i.2xlarge'|'ml.m6i.4xlarge'|'ml.m6i.8xlarge'|'ml.m6i.12xlarge'|'ml.m6i.16xlarge'|'ml.m6i.24xlarge'|'ml.m6i.32xlarge'|'ml.r6i.large'|'ml.r6i.xlarge'|'ml.r6i.2xlarge'|'ml.r6i.4xlarge'|'ml.r6i.8xlarge'|'ml.r6i.12xlarge'|'ml.r6i.16xlarge'|'ml.r6i.24xlarge'|'ml.r6i.32xlarge'|'ml.i3en.large'|'ml.i3en.xlarge'|'ml.i3en.2xlarge'|'ml.i3en.3xlarge'|'ml.i3en.6xlarge'|'ml.i3en.12xlarge'|'ml.i3en.24xlarge'|'ml.m7i.large'|'ml.m7i.xlarge'|'ml.m7i.2xlarge'|'ml.m7i.4xlarge'|'ml.m7i.8xlarge'|'ml.m7i.12xlarge'|'ml.m7i.16xlarge'|'ml.m7i.24xlarge'|'ml.m7i.48xlarge'|'ml.r7i.large'|'ml.r7i.xlarge'|'ml.r7i.2xlarge'|'ml.r7i.4xlarge'|'ml.r7i.8xlarge'|'ml.r7i.12xlarge'|'ml.r7i.16xlarge'|'ml.r7i.24xlarge'|'ml.r7i.48xlarge',
            'LifeCycleConfig': {
                'SourceS3Uri': 'string',
                'OnCreate': 'string'
            },
            'ExecutionRole': 'string',
            'ThreadsPerCore': 123,
            'InstanceStorageConfigs': [
                {
                    'EbsVolumeConfig': {
                        'VolumeSizeInGB': 123,
                        'VolumeKmsKeyId': 'string',
                        'RootVolume': True|False
                    }
                },
            ],
            'OnStartDeepHealthChecks': [
                'InstanceStress'|'InstanceConnectivity',
            ],
            'TrainingPlanArn': 'string',
            'OverrideVpcConfig': {
                'SecurityGroupIds': [
                    'string',
                ],
                'Subnets': [
                    'string',
                ]
            },
            'ScheduledUpdateConfig': {
                'ScheduleExpression': 'string',
                'DeploymentConfig': {
                    'RollingUpdatePolicy': {
                        'MaximumBatchSize': {
                            'Type': 'INSTANCE_COUNT'|'CAPACITY_PERCENTAGE',
                            'Value': 123
                        },
                        'RollbackMaximumBatchSize': {
                            'Type': 'INSTANCE_COUNT'|'CAPACITY_PERCENTAGE',
                            'Value': 123
                        }
                    },
                    'WaitIntervalInSeconds': 123,
                    'AutoRollbackConfiguration': [
                        {
                            'AlarmName': 'string'
                        },
                    ]
                }
            },
            'ImageId': 'string'
        },
    ],
    RestrictedInstanceGroups=[
        {
            'InstanceCount': 123,
            'InstanceGroupName': 'string',
            'InstanceType': 'ml.p4d.24xlarge'|'ml.p4de.24xlarge'|'ml.p5.48xlarge'|'ml.p6e-gb200.36xlarge'|'ml.trn1.32xlarge'|'ml.trn1n.32xlarge'|'ml.g5.xlarge'|'ml.g5.2xlarge'|'ml.g5.4xlarge'|'ml.g5.8xlarge'|'ml.g5.12xlarge'|'ml.g5.16xlarge'|'ml.g5.24xlarge'|'ml.g5.48xlarge'|'ml.c5.large'|'ml.c5.xlarge'|'ml.c5.2xlarge'|'ml.c5.4xlarge'|'ml.c5.9xlarge'|'ml.c5.12xlarge'|'ml.c5.18xlarge'|'ml.c5.24xlarge'|'ml.c5n.large'|'ml.c5n.2xlarge'|'ml.c5n.4xlarge'|'ml.c5n.9xlarge'|'ml.c5n.18xlarge'|'ml.m5.large'|'ml.m5.xlarge'|'ml.m5.2xlarge'|'ml.m5.4xlarge'|'ml.m5.8xlarge'|'ml.m5.12xlarge'|'ml.m5.16xlarge'|'ml.m5.24xlarge'|'ml.t3.medium'|'ml.t3.large'|'ml.t3.xlarge'|'ml.t3.2xlarge'|'ml.g6.xlarge'|'ml.g6.2xlarge'|'ml.g6.4xlarge'|'ml.g6.8xlarge'|'ml.g6.16xlarge'|'ml.g6.12xlarge'|'ml.g6.24xlarge'|'ml.g6.48xlarge'|'ml.gr6.4xlarge'|'ml.gr6.8xlarge'|'ml.g6e.xlarge'|'ml.g6e.2xlarge'|'ml.g6e.4xlarge'|'ml.g6e.8xlarge'|'ml.g6e.16xlarge'|'ml.g6e.12xlarge'|'ml.g6e.24xlarge'|'ml.g6e.48xlarge'|'ml.p5e.48xlarge'|'ml.p5en.48xlarge'|'ml.p6-b200.48xlarge'|'ml.trn2.48xlarge'|'ml.c6i.large'|'ml.c6i.xlarge'|'ml.c6i.2xlarge'|'ml.c6i.4xlarge'|'ml.c6i.8xlarge'|'ml.c6i.12xlarge'|'ml.c6i.16xlarge'|'ml.c6i.24xlarge'|'ml.c6i.32xlarge'|'ml.m6i.large'|'ml.m6i.xlarge'|'ml.m6i.2xlarge'|'ml.m6i.4xlarge'|'ml.m6i.8xlarge'|'ml.m6i.12xlarge'|'ml.m6i.16xlarge'|'ml.m6i.24xlarge'|'ml.m6i.32xlarge'|'ml.r6i.large'|'ml.r6i.xlarge'|'ml.r6i.2xlarge'|'ml.r6i.4xlarge'|'ml.r6i.8xlarge'|'ml.r6i.12xlarge'|'ml.r6i.16xlarge'|'ml.r6i.24xlarge'|'ml.r6i.32xlarge'|'ml.i3en.large'|'ml.i3en.xlarge'|'ml.i3en.2xlarge'|'ml.i3en.3xlarge'|'ml.i3en.6xlarge'|'ml.i3en.12xlarge'|'ml.i3en.24xlarge'|'ml.m7i.large'|'ml.m7i.xlarge'|'ml.m7i.2xlarge'|'ml.m7i.4xlarge'|'ml.m7i.8xlarge'|'ml.m7i.12xlarge'|'ml.m7i.16xlarge'|'ml.m7i.24xlarge'|'ml.m7i.48xlarge'|'ml.r7i.large'|'ml.r7i.xlarge'|'ml.r7i.2xlarge'|'ml.r7i.4xlarge'|'ml.r7i.8xlarge'|'ml.r7i.12xlarge'|'ml.r7i.16xlarge'|'ml.r7i.24xlarge'|'ml.r7i.48xlarge',
            'ExecutionRole': 'string',
            'ThreadsPerCore': 123,
            'InstanceStorageConfigs': [
                {
                    'EbsVolumeConfig': {
                        'VolumeSizeInGB': 123,
                        'VolumeKmsKeyId': 'string',
                        'RootVolume': True|False
                    }
                },
            ],
            'OnStartDeepHealthChecks': [
                'InstanceStress'|'InstanceConnectivity',
            ],
            'TrainingPlanArn': 'string',
            'OverrideVpcConfig': {
                'SecurityGroupIds': [
                    'string',
                ],
                'Subnets': [
                    'string',
                ]
            },
            'ScheduledUpdateConfig': {
                'ScheduleExpression': 'string',
                'DeploymentConfig': {
                    'RollingUpdatePolicy': {
                        'MaximumBatchSize': {
                            'Type': 'INSTANCE_COUNT'|'CAPACITY_PERCENTAGE',
                            'Value': 123
                        },
                        'RollbackMaximumBatchSize': {
                            'Type': 'INSTANCE_COUNT'|'CAPACITY_PERCENTAGE',
                            'Value': 123
                        }
                    },
                    'WaitIntervalInSeconds': 123,
                    'AutoRollbackConfiguration': [
                        {
                            'AlarmName': 'string'
                        },
                    ]
                }
            },
            'EnvironmentConfig': {
                'FSxLustreConfig': {
                    'SizeInGiB': 123,
                    'PerUnitStorageThroughput': 123
                }
            }
        },
    ],
    NodeRecovery='Automatic'|'None',
    InstanceGroupsToDelete=[
        'string',
    ],
    ClusterRole='string',
    AutoScaling={
        'Mode': 'Enable'|'Disable',
        'AutoScalerType': 'Karpenter'
    }
)
type ClusterName:

string

param ClusterName:

[REQUIRED]

Specify the name of the SageMaker HyperPod cluster you want to update.

type InstanceGroups:

list

param InstanceGroups:

Specify the instance groups to update.

  • (dict) --

    The specifications of an instance group that you need to define.

    • InstanceCount (integer) -- [REQUIRED]

      Specifies the number of instances to add to the instance group of a SageMaker HyperPod cluster.

    • InstanceGroupName (string) -- [REQUIRED]

      Specifies the name of the instance group.

    • InstanceType (string) -- [REQUIRED]

      Specifies the instance type of the instance group.

    • LifeCycleConfig (dict) -- [REQUIRED]

      Specifies the LifeCycle configuration for the instance group.

      • SourceS3Uri (string) -- [REQUIRED]

        An Amazon S3 bucket path where your lifecycle scripts are stored.

      • OnCreate (string) -- [REQUIRED]

        The file name of the entrypoint script of lifecycle scripts under SourceS3Uri. This entrypoint script runs during cluster creation.

    • ExecutionRole (string) -- [REQUIRED]

      Specifies an IAM execution role to be assumed by the instance group.

    • ThreadsPerCore (integer) --

      Specifies the value for Threads per core. For instance types that support multithreading, you can specify 1 for disabling multithreading and 2 for enabling multithreading. For instance types that doesn't support multithreading, specify 1. For more information, see the reference table of CPU cores and threads per CPU core per instance type in the Amazon Elastic Compute Cloud User Guide.

    • InstanceStorageConfigs (list) --

      Specifies the additional storage configurations for the instances in the SageMaker HyperPod cluster instance group.

      • (dict) --

        Defines the configuration for attaching additional storage to the instances in the SageMaker HyperPod cluster instance group. To learn more, see SageMaker HyperPod release notes: June 20, 2024.

        • EbsVolumeConfig (dict) --

          Defines the configuration for attaching additional Amazon Elastic Block Store (EBS) volumes to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to /opt/sagemaker.

          • VolumeSizeInGB (integer) --

            The size in gigabytes (GB) of the additional EBS volume to be attached to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to /opt/sagemaker.

          • VolumeKmsKeyId (string) --

            The ID of a KMS key to encrypt the Amazon EBS volume.

          • RootVolume (boolean) --

            Specifies whether the configuration is for the cluster's root or secondary Amazon EBS volume. You can specify two ClusterEbsVolumeConfig fields to configure both the root and secondary volumes. Set the value to True if you'd like to provide your own customer managed Amazon Web Services KMS key to encrypt the root volume. When True:

            • The configuration is applied to the root volume.

            • You can't specify the VolumeSizeInGB field. The size of the root volume is determined for you.

            • You must specify a KMS key ID for VolumeKmsKeyId to encrypt the root volume with your own KMS key instead of an Amazon Web Services owned KMS key.

            Otherwise, by default, the value is False, and the following applies:

            • The configuration is applied to the secondary volume, while the root volume is encrypted with an Amazon Web Services owned key.

            • You must specify the VolumeSizeInGB field.

            • You can optionally specify the VolumeKmsKeyId to encrypt the secondary volume with your own KMS key instead of an Amazon Web Services owned KMS key.

    • OnStartDeepHealthChecks (list) --

      A flag indicating whether deep health checks should be performed when the cluster instance group is created or updated.

      • (string) --

    • TrainingPlanArn (string) --

      The Amazon Resource Name (ARN); of the training plan to use for this cluster instance group.

      For more information about how to reserve GPU capacity for your SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see ``CreateTrainingPlan ``.

    • OverrideVpcConfig (dict) --

      To configure multi-AZ deployments, customize the Amazon VPC configuration at the instance group level. You can specify different subnets and security groups across different AZs in the instance group specification to override a SageMaker HyperPod cluster's default Amazon VPC configuration. For more information about deploying a cluster in multiple AZs, see Setting up SageMaker HyperPod clusters across multiple AZs.

      • SecurityGroupIds (list) -- [REQUIRED]

        The VPC security group IDs, in the form sg-xxxxxxxx. Specify the security groups for the VPC that is specified in the Subnets field.

        • (string) --

      • Subnets (list) -- [REQUIRED]

        The ID of the subnets in the VPC to which you want to connect your training job or model. For information about the availability of specific instance types, see Supported Instance Types and Availability Zones.

        • (string) --

    • ScheduledUpdateConfig (dict) --

      The configuration object of the schedule that SageMaker uses to update the AMI.

      • ScheduleExpression (string) -- [REQUIRED]

        A cron expression that specifies the schedule that SageMaker follows when updating the AMI.

      • DeploymentConfig (dict) --

        The configuration to use when updating the AMI versions.

        • RollingUpdatePolicy (dict) --

          The policy that SageMaker uses when updating the AMI versions of the cluster.

          • MaximumBatchSize (dict) -- [REQUIRED]

            The maximum amount of instances in the cluster that SageMaker can update at a time.

            • Type (string) -- [REQUIRED]

              Specifies whether SageMaker should process the update by amount or percentage of instances.

            • Value (integer) -- [REQUIRED]

              Specifies the amount or percentage of instances SageMaker updates at a time.

          • RollbackMaximumBatchSize (dict) --

            The maximum amount of instances in the cluster that SageMaker can roll back at a time.

            • Type (string) -- [REQUIRED]

              Specifies whether SageMaker should process the update by amount or percentage of instances.

            • Value (integer) -- [REQUIRED]

              Specifies the amount or percentage of instances SageMaker updates at a time.

        • WaitIntervalInSeconds (integer) --

          The duration in seconds that SageMaker waits before updating more instances in the cluster.

        • AutoRollbackConfiguration (list) --

          An array that contains the alarms that SageMaker monitors to know whether to roll back the AMI update.

          • (dict) --

            The details of the alarm to monitor during the AMI update.

            • AlarmName (string) -- [REQUIRED]

              The name of the alarm.

    • ImageId (string) --

      When configuring your HyperPod cluster, you can specify an image ID using one of the following options:

      • HyperPodPublicAmiId: Use a HyperPod public AMI

      • CustomAmiId: Use your custom AMI

      • default: Use the default latest system image

      If you choose to use a custom AMI ( CustomAmiId), ensure it meets the following requirements:

      • Encryption: The custom AMI must be unencrypted.

      • Ownership: The custom AMI must be owned by the same Amazon Web Services account that is creating the HyperPod cluster.

      • Volume support: Only the primary AMI snapshot volume is supported; additional AMI volumes are not supported.

      When updating the instance group's AMI through the UpdateClusterSoftware operation, if an instance group uses a custom AMI, you must provide an ImageId or use the default as input. Note that if you don't specify an instance group in your UpdateClusterSoftware request, then all of the instance groups are patched with the specified image.

type RestrictedInstanceGroups:

list

param RestrictedInstanceGroups:

The specialized instance groups for training models like Amazon Nova to be created in the SageMaker HyperPod cluster.

  • (dict) --

    The specifications of a restricted instance group that you need to define.

    • InstanceCount (integer) -- [REQUIRED]

      Specifies the number of instances to add to the restricted instance group of a SageMaker HyperPod cluster.

    • InstanceGroupName (string) -- [REQUIRED]

      Specifies the name of the restricted instance group.

    • InstanceType (string) -- [REQUIRED]

      Specifies the instance type of the restricted instance group.

    • ExecutionRole (string) -- [REQUIRED]

      Specifies an IAM execution role to be assumed by the restricted instance group.

    • ThreadsPerCore (integer) --

      The number you specified to TreadsPerCore in CreateCluster for enabling or disabling multithreading. For instance types that support multithreading, you can specify 1 for disabling multithreading and 2 for enabling multithreading. For more information, see the reference table of CPU cores and threads per CPU core per instance type in the Amazon Elastic Compute Cloud User Guide.

    • InstanceStorageConfigs (list) --

      Specifies the additional storage configurations for the instances in the SageMaker HyperPod cluster restricted instance group.

      • (dict) --

        Defines the configuration for attaching additional storage to the instances in the SageMaker HyperPod cluster instance group. To learn more, see SageMaker HyperPod release notes: June 20, 2024.

        • EbsVolumeConfig (dict) --

          Defines the configuration for attaching additional Amazon Elastic Block Store (EBS) volumes to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to /opt/sagemaker.

          • VolumeSizeInGB (integer) --

            The size in gigabytes (GB) of the additional EBS volume to be attached to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to /opt/sagemaker.

          • VolumeKmsKeyId (string) --

            The ID of a KMS key to encrypt the Amazon EBS volume.

          • RootVolume (boolean) --

            Specifies whether the configuration is for the cluster's root or secondary Amazon EBS volume. You can specify two ClusterEbsVolumeConfig fields to configure both the root and secondary volumes. Set the value to True if you'd like to provide your own customer managed Amazon Web Services KMS key to encrypt the root volume. When True:

            • The configuration is applied to the root volume.

            • You can't specify the VolumeSizeInGB field. The size of the root volume is determined for you.

            • You must specify a KMS key ID for VolumeKmsKeyId to encrypt the root volume with your own KMS key instead of an Amazon Web Services owned KMS key.

            Otherwise, by default, the value is False, and the following applies:

            • The configuration is applied to the secondary volume, while the root volume is encrypted with an Amazon Web Services owned key.

            • You must specify the VolumeSizeInGB field.

            • You can optionally specify the VolumeKmsKeyId to encrypt the secondary volume with your own KMS key instead of an Amazon Web Services owned KMS key.

    • OnStartDeepHealthChecks (list) --

      A flag indicating whether deep health checks should be performed when the cluster restricted instance group is created or updated.

      • (string) --

    • TrainingPlanArn (string) --

      The Amazon Resource Name (ARN) of the training plan to filter clusters by. For more information about reserving GPU capacity for your SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see ``CreateTrainingPlan ``.

    • OverrideVpcConfig (dict) --

      Specifies an Amazon Virtual Private Cloud (VPC) that your SageMaker jobs, hosted models, and compute resources have access to. You can control access to and from your resources by configuring a VPC. For more information, see Give SageMaker Access to Resources in your Amazon VPC.

      • SecurityGroupIds (list) -- [REQUIRED]

        The VPC security group IDs, in the form sg-xxxxxxxx. Specify the security groups for the VPC that is specified in the Subnets field.

        • (string) --

      • Subnets (list) -- [REQUIRED]

        The ID of the subnets in the VPC to which you want to connect your training job or model. For information about the availability of specific instance types, see Supported Instance Types and Availability Zones.

        • (string) --

    • ScheduledUpdateConfig (dict) --

      The configuration object of the schedule that SageMaker follows when updating the AMI.

      • ScheduleExpression (string) -- [REQUIRED]

        A cron expression that specifies the schedule that SageMaker follows when updating the AMI.

      • DeploymentConfig (dict) --

        The configuration to use when updating the AMI versions.

        • RollingUpdatePolicy (dict) --

          The policy that SageMaker uses when updating the AMI versions of the cluster.

          • MaximumBatchSize (dict) -- [REQUIRED]

            The maximum amount of instances in the cluster that SageMaker can update at a time.

            • Type (string) -- [REQUIRED]

              Specifies whether SageMaker should process the update by amount or percentage of instances.

            • Value (integer) -- [REQUIRED]

              Specifies the amount or percentage of instances SageMaker updates at a time.

          • RollbackMaximumBatchSize (dict) --

            The maximum amount of instances in the cluster that SageMaker can roll back at a time.

            • Type (string) -- [REQUIRED]

              Specifies whether SageMaker should process the update by amount or percentage of instances.

            • Value (integer) -- [REQUIRED]

              Specifies the amount or percentage of instances SageMaker updates at a time.

        • WaitIntervalInSeconds (integer) --

          The duration in seconds that SageMaker waits before updating more instances in the cluster.

        • AutoRollbackConfiguration (list) --

          An array that contains the alarms that SageMaker monitors to know whether to roll back the AMI update.

          • (dict) --

            The details of the alarm to monitor during the AMI update.

            • AlarmName (string) -- [REQUIRED]

              The name of the alarm.

    • EnvironmentConfig (dict) -- [REQUIRED]

      The configuration for the restricted instance groups (RIG) environment.

      • FSxLustreConfig (dict) --

        Configuration settings for an Amazon FSx for Lustre file system to be used with the cluster.

        • SizeInGiB (integer) -- [REQUIRED]

          The storage capacity of the Amazon FSx for Lustre file system, specified in gibibytes (GiB).

        • PerUnitStorageThroughput (integer) -- [REQUIRED]

          The throughput capacity of the Amazon FSx for Lustre file system, measured in MB/s per TiB of storage.

type NodeRecovery:

string

param NodeRecovery:

The node recovery mode to be applied to the SageMaker HyperPod cluster.

type InstanceGroupsToDelete:

list

param InstanceGroupsToDelete:

Specify the names of the instance groups to delete. Use a single , as the separator between multiple names.

  • (string) --

type ClusterRole:

string

param ClusterRole:

The Amazon Resource Name (ARN) of the IAM role that HyperPod assumes for cluster autoscaling operations. Cannot be updated while autoscaling is enabled.

type AutoScaling:

dict

param AutoScaling:

Updates the autoscaling configuration for the cluster. Use to enable or disable automatic node scaling.

  • Mode (string) -- [REQUIRED]

    Describes whether autoscaling is enabled or disabled for the cluster. Valid values are Enable and Disable.

  • AutoScalerType (string) --

    The type of autoscaler to use. Currently supported value is Karpenter.

rtype:

dict

returns:

Response Syntax

{
    'ClusterArn': 'string'
}

Response Structure

  • (dict) --

    • ClusterArn (string) --

      The Amazon Resource Name (ARN) of the updated SageMaker HyperPod cluster.