View a markdown version of this page

Crea un SageMaker HyperPod cluster sui piani di formazione utilizzando l' SageMaker API, oppure AWS CLI - Amazon SageMaker AI

Le traduzioni sono generate tramite traduzione automatica. In caso di conflitto tra il contenuto di una traduzione e la versione originale in Inglese, quest'ultima prevarrà.

Crea un SageMaker HyperPod cluster sui piani di formazione utilizzando l' SageMaker API, oppure AWS CLI

Per utilizzare i piani di SageMaker formazione per il tuo SageMaker HyperPod cluster Amazon, specifica l'ARN del piano di formazione che desideri utilizzare nel TrainingPlanArn parametro di ClusterInstanceGroupSpecification quando richiami l'CreateClusteroperazione API.

Assicurati che la sottorete associata all’AZ indicata del piano sia inclusa in VPCConfig della tua configurazione del cluster. Puoi recuperare l'importo AvailabilityZone di un piano di formazione nella risposta a una chiamata DescribeTrainingPlan API.

L'esempio seguente illustra come creare un nuovo SageMaker HyperPod cluster e fornire a un gruppo di istanze un piano di formazione nell'--instance-groupsattributo del create-cluster AWS CLI comando. 

# 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,\
        }]'

Per informazioni su come creare un HyperPod cluster utilizzando il AWS CLI, vedere create-cluster.

Dopo aver creato il cluster, puoi verificare che al gruppo di istanze sia stata assegnata correttamente la capacità prevista dal piano di addestramento chiamando l’API DescribeCluster.

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