Amazon SageMaker HyperPod 现已支持适用于使用持续预置的 Slurm 集群的基于 AMI 的节点生命周期配置
Amazon SageMaker HyperPod 现已支持适用于使用持续预置的 Slurm 集群的基于 AMI 的节点生命周期配置。持续预置会在容量可用时向集群添加节点,通过此次发布,基于 AMI 的配置将扩展到使用此模式的集群。有了这一支持,无需下载、配置生命周期配置脚本或将该脚本上传到 Amazon S3,即可创建使用持续预置的集群。
基于 AMI 的配置将为节点提供生产就绪型环境所需的软件和配置,以运行人工智能/机器学习训练工作负载,包括 Docker、Enroot 和 Pyxis 等所需的软件,以及 Slurm 会计、SSH 密钥生成和日志轮换等配置。在使用持续预置时,每个节点均是在添加到集群时通过 AMI 配置的,无需管理生命周期配置脚本,这使得节点能够更快地用于调度作业。要启用基于 AMI 的配置,请在通过 API 创建集群时,从实例组配置中省略 LifeCycleConfig 块,或者在使用 SageMaker AI 控制台时,在“自定义设置”的“生命周期脚本”下选择“无”。要在基于 AMI 的配置基准的基础上进行更多自定义设置,可提供扩展脚本,方式是通过 API 在 LifecycleConfig 块中指定 OnInitComplete 参数和 SourceS3Uri,或在使用控制台时在“自定义设置”的“S3 中的扩展脚本文件”字段中提供 S3 URI。对于需要完全控制预置的使用案例,仍完全支持自定义生命周期配置脚本。
适用于使用持续预置的 Slurm 集群的基于 AMI 的节点生命周期配置已在提供 SageMaker HyperPod 的所有 AWS 区域推出。要开始使用,请参阅 SageMaker 用户指南中的 Getting started with SageMaker HyperPod using the AWS CLI 或 Getting started with SageMaker HyperPod using the SageMaker AI console。