准备和上传生命周期脚本 - 亚马逊 SageMaker AI
准备生命周期脚本创建配置文件验证 Amazon S3 存储桶中的文件
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异，请参阅 中国的 Amazon Web Services 服务入门 (PDF)。

本文属于机器翻译版本。若本译文内容与英语原文存在差异，则一律以英文原文为准。

准备和上传生命周期脚本

创建完所有必需的资源后，您需要为 SageMaker HyperPod 集群设置生命周期脚本。这些生命周期脚本提供了可用于创建基本 HyperPod Slurm 集群的基本配置。

准备生命周期脚本

按照以下步骤获取生命周期脚本。

  1. 将生命周期脚本从 GitHub 存储库下载到您的计算机。

  2. 使用 c p C LI 命令将生命周期脚本上传到您在中配置基本资源创建的 Amazon S3 存储桶。

    aws s3 cp --recursive LifeCycleScripts/base-config s3://${ROOT_BUCKET_NAME}/LifeCycleScripts/base-config

创建配置文件

按照以下步骤创建配置文件并将其上传到存储生命周期脚本的同一 Amazon S3 存储桶。

  1. 使用以下配置创建名provisioning_parameters.json为的配置文件。请注意，slurm_sns_arn 是可选项。如果未提供， HyperPod 则无法设置 Amazon SNS 通知。

    cat <<EOF > /tmp/provisioning_parameters.json
{
  "version": "1.0.0",
  "workload_manager": "slurm",
  "controller_group": "$CONTOLLER_IG_NAME",
  "login_group": "my-login-group",
  "worker_groups": [
    {
      "instance_group_name": "$COMPUTE_IG_NAME",
      "partition_name": "dev"
    }
  ],
  "fsx_dns_name": "$SLURM_FSX_DNS_NAME",
  "fsx_mountname": "$SLURM_FSX_MOUNT_NAME",
  "slurm_configurations": {
    "slurm_database_secret_arn": "$SLURM_DB_SECRET_ARN",
    "slurm_database_endpoint": "$SLURM_DB_ENDPOINT_ADDRESS",
    "slurm_shared_directory": "/fsx",
    "slurm_database_user": "$DB_USER_NAME",
    "slurm_sns_arn": "$SLURM_SNS_FAILOVER_TOPIC_ARN"
  }
}
EOF

  2. 将provisioning_parameters.json文件上传到存储生命周期脚本的同一 Amazon S3 存储桶。

    aws s3 cp /tmp/provisioning_parameters.json s3://${ROOT_BUCKET_NAME}/LifeCycleScripts/base-config/provisioning_parameters.json

验证 Amazon S3 存储桶中的文件

上传所有生命周期脚本和provisioning_parameters.json文件后，您的 Amazon S3 存储桶应如下所示。

该图显示了在亚马逊简单存储服务控制台中上传到 Amazon S3 存储桶的所有生命周期脚本。

有关更多信息，请参阅从提供的基本生命周期脚本开始 HyperPod。