使用 AWS CLI

准备生命周期配置脚本并将其上传到 S3 存储桶，如 s3://sagemaker-amzn-s3-demo-bucket/lifecycle-script-directory/src/。下面的步骤 2 假设在指定的 S3 存储桶中有一个名为 on_create.sh 的入口点脚本。

准备一个 JSON 格式的 CreateClusterAPI 请求文件。您应配置实例组，使其与您在 provisioning_params.json 文件中设计的 Slurm 集群相匹配，该文件将在创建集群时使用，是运行一组生命周期脚本的一部分。要了解更多信息，请参阅 使用生命周期脚本自定义 SageMaker HyperPod集群。下面的模板有两个实例组，以满足 Slurm 集群的最低要求：一个控制器（主）节点和一个计算（Worker）节点。对于 ExecutionRole，请提供使用托管的 AmazonSageMakerClusterInstanceRolePolicy 从 的 IAM 角色适用于 SageMaker HyperPod 部分创建的 IAM 角色的 ARN。

// create_cluster.json
{
    "ClusterName": "your-hyperpod-cluster",
    "InstanceGroups": [
        {
            "InstanceGroupName": "controller-group",
            "InstanceType": "ml.m5.xlarge",
            "InstanceCount": 1,
            "LifeCycleConfig": {
                "SourceS3Uri": "s3://amzn-s3-demo-bucket-sagemaker/lifecycle-script-directory/src/",
                "OnCreate": "on_create.sh"
            },
            "ExecutionRole": "arn:aws:iam::111122223333:role/iam-role-for-cluster",
            // Optional: Configure an additional storage per instance group.
            "InstanceStorageConfigs": [
                {
                   // Attach an additional EBS volume to each instance within the instance group.
                   // The default mount path for the additional EBS volume is /opt/sagemaker.
                   "EbsVolumeConfig":{
                      // Specify an integer between 1 and 16384 in gigabytes (GB).
                      "VolumeSizeInGB": integer,
                   }
                }
            ]
        }, 
        {
            "InstanceGroupName": "worker-group-1",
            "InstanceType": "ml.p4d.xlarge",
            "InstanceCount": 1,
            "LifeCycleConfig": {
                "SourceS3Uri": "s3://amzn-s3-demo-bucket-sagemaker/lifecycle-script-directory/src/",
                "OnCreate": "on_create.sh"
            },
            "ExecutionRole": "arn:aws:iam::111122223333:role/iam-role-for-cluster"
        }
    ],
    // Optional
    "Tags": [ 
        { 
           "Key": "string",
           "Value": "string"
        }
    ],
    // Optional
    "VpcConfig": { 
        "SecurityGroupIds": [ "string" ],
        "Subnets": [ "string" ]
    }
}

根据生命周期脚本设计集群结构的方式，在 InstanceGroups 参数下最多可配置 20 个实例组。

对于Tags请求参数，您可以添加用于将 SageMaker HyperPod 集群作为 AWS 资源进行管理的自定义标签。您可以像在其他支持标记的 AWS 服务中添加标签一样向集群添加标签。要详细了解如何为 AWS 资源添加标签，请参阅《标记 AWS 资源用户指南》。

对于 VpcConfig 请求参数，请指定要使用的 VPC 信息。有关更多信息，请参阅 SageMaker HyperPod使用自定义 Amazon VPC 进行设置。

运行 create-cluster 命令如下。

aws sagemaker create-cluster \
    --cli-input-json file://complete/path/to/create_cluster.json

这将返回新集群的 ARN。