Crea un SageMaker HyperPod clúster sobre los planes de formación mediante la SageMaker API, o AWS CLI - Amazon SageMaker AI

Las traducciones son generadas a través de traducción automática. En caso de conflicto entre la traducción y la version original de inglés, prevalecerá la version en inglés.

Crea un SageMaker HyperPod clúster sobre los planes de formación mediante la SageMaker API, o AWS CLI

Para usar planes de SageMaker entrenamiento para tu SageMaker HyperPod clúster de Amazon, especifica el ARN del plan de entrenamiento que deseas usar en el TrainingPlanArnparámetro de la operación ClusterInstanceGroupSpecificational llamar a la CreateClusterAPI.

Asegúrese de que la subred asociada a la zona de disponibilidad designada de su plan esté incluida en la VPCConfig de la configuración de su clúster. Puedes recuperar el plan AvailabilityZone de formación en respuesta a una llamada a la DescribeTrainingPlanAPI.

En el siguiente ejemplo, se muestra cómo crear un SageMaker HyperPod clúster nuevo y proporcionar a un grupo de instancias un plan de entrenamiento en el --instance-groups atributo 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,\
        }]'

Para obtener información sobre cómo crear un HyperPod clúster con el AWS CLI, consulte create-cluster.

Tras crear el clúster, puede comprobar que se haya asignado capacidad del plan de entrenamiento a su grupo de instancias llamando a la API DescribeCluster.

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