使用 管理 SageMaker HyperPod Slurm 叢集 AWS CLI

準備生命週期組態指令碼，並將其上傳至 S3 儲存貯體，例如 s3://sagemaker-amzn-s3-demo-bucket/lifecycle-script-directory/src/。下列步驟 2 假設在指定的 S3 儲存貯體中有名為 on_create.sh 的進入點指令碼。

準備 JSON 格式的 CreateCluster API 請求檔案。您應該將執行個體群組設定為與您在 provisioning_parameters.json 檔案中設計的 Slurm 叢集相符，此檔案會在叢集建立期間用作執行一組生命週期指令碼的一部分。如需詳細資訊，請參閱 使用生命週期指令碼自訂 SageMaker HyperPod 叢集。下列範本有兩個執行個體群組，以符合 Slurm 叢集的最低要求：一個控制器 (主) 節點和一個運算 (工作) 節點。針對 ExecutionRole，請從 SageMaker HyperPod 的 IAM 角色 區段提供您使用受管 AmazonSageMakerClusterInstanceRolePolicy 建立的 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 資訊。如需詳細資訊，請參閱使用自訂 Amazon VPC 設定 SageMaker HyperPod。

執行 create-cluster 命令，如下所示。

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

這應該會傳回新叢集的 ARN。