Amazon SageMaker HyperPod で Slurm クラスターの AMI ベースのノードライフサイクル設定が継続的プロビジョニングでサポート

Amazon SageMaker HyperPod が、継続的プロビジョニングを使用する Slurm クラスター向けに AMI ベースのノードライフサイクル設定をサポートするようになった。継続的プロビジョニングでは、容量が確保されるときにクラスターにノードを追加する。このサポートにより、ライフサイクル設定スクリプトを Amazon S3 にダウンロード、構成、またはアップロードすることなく、継続的プロビジョニングを使用するクラスターを作成できる。AMI ベースの構成では、Docker、Enroot、Pyxis などの必要なソフトウェアや、Slurm アカウンティング、SSH キー生成、ログローテーションなどの構成で、ノードをプロダクション対応環境で実行できるようにする。継続的プロビジョニングを使用する場合、各ノードはクラスターに追加されるときに AMI から構成され、ライフサイクル設定スクリプトを管理する必要がなく、ジョブのスケジューリングがより迅速になる。AMI ベースの構成を有効にするには、インスタンスグループ構成の LifeCycleConfig ブロックを省略するか、SageMaker AI コンソールでカスタム設定のライフサイクルスクリプトで「なし」を選択する。AMI ベースの構成ベースでさらにカスタマイズが必要な場合は、LifeCycleConfig ブロックの OnInitComplete パラメータと SourceS3Uri を指定するか、コンソールでカスタム設定の「S3 の拡張スクリプトファイル」フィールドに S3 URI を提供することで、拡張スクリプトを提供できる。完全なプロビジョニングコントロールが必要なユースケースには、カスタムライフサイクル設定スクリプトが引き続き完全にサポートされている。この機能は、SageMaker HyperPod が利用可能なすべての AWS リージョンで利用できる。