SageMaker API を使用してトレーニングプランで SageMaker HyperPod クラスターを作成する、または AWS CLI

Amazon SageMaker HyperPod クラスターで SageMaker トレーニングプランを使用するには、 CreateCluster API オペレーションを呼び出すClusterInstanceGroupSpecificationときに、 の TrainingPlanArnパラメータで使用するトレーニングプランの ARN を指定します。 Amazon SageMaker

プランの指定された AZ に関連付けられたサブネットが、クラスター設定VPCConfigの に含まれていることを確認します。DescribeTrainingPlan API コールに応答して、トレーニングプランAvailabilityZoneの を取得できます。

次のサンプルは、新しい SageMaker HyperPod クラスターを作成し、 create-cluster AWS CLI コマンドの --instance-groups 属性でインスタンスグループにトレーニングプランを提供する方法を示しています。

# Create a cluster         
aws sagemaker create-cluster \
  --cluster-name cluster-name \
  --instance-groups '[ \
        { \
            "InstanceCount": 1,\
            "InstanceGroupName": "controller-nodes",\
            "InstanceType": "ml.t3.xlarge",\
            "LifeCycleConfig": {"SourceS3Uri": source_s3_uri, "OnCreate": "on_create.sh"},\
            "ExecutionRole": "arn:aws:iam::customer_account_id:role/execution_role",\
            "ThreadsPerCore": 1,\
        },\
        { \
            "InstanceCount": 2, \
            "InstanceGroupName": "worker-nodes",\
            "InstanceType": "p4d.24xlarge",\
            "LifeCycleConfig": {"SourceS3Uri": source_s3_uri, "OnCreate": "on_create.sh"},\
            "ExecutionRole": "arn:aws:iam::customer_account_id}:role/execution_role}",\
            "ThreadsPerCore": 1,\
            "TrainingPlanArn": training_plan_arn,\
        }]'

を使用して HyperPod クラスターを作成する方法については AWS CLI、「」を参照してくださいcreate-cluster。

クラスターを作成したら、 DescribeCluster API を呼び出して、インスタンスグループがトレーニングプランから適切にキャパシティーに割り当てられたことを確認できます。

aws sagemaker describe-cluster --cluster-name cluster-name