使用 的 AWS CLI 命令 SageMaker HyperPod APIs

首先，準備生命週期指令碼並將其上傳至 Amazon S3 儲存貯體。在叢集建立期間， 會在每個執行個體群組中 HyperPod 執行它們。使用下列命令將生命週期指令碼上傳至 Amazon S3。

aws s3 sync \
    ~/local-dir-to-lifecycle-scripts/* \ 
    s3://sagemaker-<unique-s3-bucket-name>/<lifecycle-script-directory>/src

如果您是從頭開始，請使用 Awsome 分散式訓練 GitHub 儲存庫 中提供的範例生命週期指令碼。下列子步驟說明如何下載、修改內容，以及如何將範例生命週期指令碼上傳至 Amazon S3 儲存貯體。

1. 將生命週期指令碼範例的複本下載至本機電腦上的目錄。

   git clone https://github.com/aws-samples/awsome-distributed-training/

2. 前往目錄 1.architectures/5.sagemaker_hyperpods/LifecycleScripts/base-config，您可以在其中找到一組生命週期指令碼。

   cd awsome-distributed-training/1.architectures/5.sagemaker_hyperpods/LifecycleScripts/base-config

   若要進一步了解生命週期指令碼範例，請參閱 使用生命週期指令碼自訂 SageMaker HyperPod叢集。

3. 寫入 Slurm 組態檔案，並將其儲存為 provisioning_params.json。在 檔案中，指定基本 Slurm 組態參數，以將 Slurm 節點正確指派給 SageMaker HyperPod 叢集執行個體群組。在本教學課程中，設定三個名為 my-controller-group、 my-login-group和 的 Slurm 節點worker-group-1，如下列範例組態 所示provisioning_params.json。

   {
       "version": "1.0.0",
       "workload_manager": "slurm",
       "controller_group": "my-controller-group",
       "login_group": "my-login-group",
       "worker_groups": [
           {
               "instance_group_name": "worker-group-1",
               "partition_name": "partition-1"
           }
       ]
   }

4. 將指令碼上傳至 s3://sagemaker-<unique-s3-bucket-name>/<lifecycle-script-directory>/src。您可以使用 Amazon S3 主控台，或執行下列 AWS CLI Amazon S3 命令來執行此操作。

   aws s3 sync \
       ~/local-dir-to-lifecycle-scripts/* \ 
       s3://sagemaker-<unique-s3-bucket-name>/<lifecycle-script-directory>/src

準備 JSON 格式的CreateCluster請求檔案，並儲存為 create_cluster.json。下列請求範本與步驟 1.c provisioning_params.json 中定義的 Slurm 節點組態相符。針對 ExecutionRole，提供您在 AmazonSageMakerClusterInstanceRolePolicy中使用 受管 所建立IAM角色ARN的 使用 SageMaker HyperPod 的先決條件。

{
    // Required: Specify the name of the cluster.
    "ClusterName": "my-hyperpod-cluster",
    // Required: Configure instance groups to be launched in the cluster
    "InstanceGroups": [
        {
            // Required: Specify the basic configurations to set up a controller node.
            "InstanceGroupName": "my-controller-group",
            "InstanceType": "ml.c5.xlarge",
            "InstanceCount": 1,
            "LifeCycleConfig": {
              "SourceS3Uri": "s3://sagemaker-<unique-s3-bucket-name>/<lifecycle-script-directory>/src",
              "OnCreate": "on_create.sh"
            },
            "ExecutionRole": "${ROLE}",
            // 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": "my-login-group",
            "InstanceType": "ml.m5.4xlarge",
            "InstanceCount": 1,
            "LifeCycleConfig": {
              "SourceS3Uri": "s3://sagemaker-<unique-s3-bucket-name>/<lifecycle-script-directory>/src",
              "OnCreate": "on_create.sh"
            },
            "ExecutionRole": "${ROLE}"
        },
        {
            "InstanceGroupName": "worker-group-1",
            "InstanceType": "ml.trn1.32xlarge",
            "InstanceCount": 1,
            "LifeCycleConfig": {
              "SourceS3Uri": "s3://sagemaker-<unique-s3-bucket-name>/<lifecycle-script-directory>/src",
              "OnCreate": "on_create.sh"
            },
            "ExecutionRole": "${ROLE}"
        }
    ]
}

執行下列命令來建立叢集。

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

這應該會傳回已建立叢集ARN的 。

如果您因為資源限制而收到錯誤，請確定您將執行個體類型變更為帳戶中具有足夠配額的執行個體類型，或遵循 的 請求額外的配額SageMaker HyperPod 配額。

執行 describe-cluster 以檢查叢集的狀態。

aws sagemaker describe-cluster --cluster-name my-hyperpod-cluster

叢集的狀態變為 後InService，繼續下一個步驟。

執行 list-cluster-nodes 以檢查叢集節點的詳細資訊。

aws sagemaker list-cluster-nodes --cluster-name my-hyperpod-cluster

這會傳回回應，而 InstanceId是您的叢集使用者記錄 （aws ssm） 到其中所需的 。如需登入叢集節點和執行 ML 工作負載的詳細資訊，請參閱 SageMaker HyperPod 叢集上的任務。