사용 AWS CLI

수명 주기 구성 스크립트를 준비하고 와 같은 S3 버킷에 업로드합니다s3://sagemaker-<amzn-s3-demo-bucket>/<lifecycle-script-directory>/src/. 다음 2단계에서는 지정된 S3 버킷on_create.sh에 라는 진입점 스크립트가 있다고 가정합니다.

JSON 형식의 CreateCluster API 요청 파일을 준비합니다. 수명 주기 스크립트 세트 실행의 일부로 클러스터 생성 중에 사용할 provisioning_params.json 파일에서 설계한 Slurm 클러스터와 일치하도록 인스턴스 그룹을 구성해야 합니다. 자세한 내용은 수명 주기 스크립트를 사용하여 클러스터 사용자 지정 SageMaker HyperPod 을 참조하십시오. 다음 템플릿에는 Slurm 클러스터의 최소 요구 사항을 충족하는 두 개의 인스턴스 그룹이 있습니다. 컨트롤러(헤드) 노드 1개와 컴퓨팅(워커) 노드 1개. 의 경우 섹션에서 관리형 로 생성한 IAM 역할ARN의 AmazonSageMakerClusterInstanceRolePolicy를 ExecutionRole제공합니다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의 가 반환됩니다.