の使用 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 クラスターの最小要件を満たすための 2 つのインスタンスグループがあります。1 つのコントローラー (ヘッド) ノードと 1 つのコンピューティング (ワーカー) ノードです。にはExecutionRole、 セクション AmazonSageMakerClusterInstanceRolePolicy から で管理された で作成したIAMロールARNの を指定しますIAM の ロール SageMaker HyperPod。

// create_cluster.json
{
    "ClusterName": "your-hyperpod-cluster",
    "InstanceGroups": [
        {
            "InstanceGroupName": "controller-group",
            "InstanceType": "ml.m5.xlarge",
            "InstanceCount": 1,
            "LifeCycleConfig": {
                "SourceS3Uri": "s3://sagemaker-<amzn-s3-demo-bucket>/<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://sagemaker-<amzn-s3-demo-bucket>/<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 でのセットアップ SageMaker HyperPod VPC」を参照してください。

create-cluster コマンドを次のように実行します。

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

これにより、新しいクラスターARNの が返されます。