Amazon SageMaker HyperPod prend désormais en charge la configuration du cycle de vie des nœuds basée sur AMI pour les clusters Slurm à l’aide du provisionnement continu
Amazon SageMaker HyperPod prend désormais en charge la configuration basée sur l’AMI pour les clusters Slurm qui utilisent le provisionnement continu. Le provisionnement continu ajoute des nœuds au cluster à mesure que la capacité devient disponible, et ce lancement étend la configuration basée sur l’AMI aux clusters utilisant ce mode. Grâce à cette prise en charge, des clusters utilisant le provisionnement continu peuvent être créés sans télécharger, configurer ni charger des scripts de configuration du cycle de vie sur Amazon S3.
La configuration basée sur l’AMI fournit aux nœuds les logiciels et les configurations nécessaires à un environnement prêt à la production pour exécuter les charges de travail de formation IA/ML, y compris les logiciels requis tels que Docker, Enroot et Pyxis, et des configurations telles que la comptabilité Slurm, la génération de clés SSH et la rotation des journaux. Lors de l’utilisation du provisionnement continu, chaque nœud est configuré à partir de l’AMI au fur et à mesure de son ajout au cluster, sans qu’il soit nécessaire de gérer les scripts de configuration du cycle de vie, de sorte que les nœuds deviennent disponibles pour planifier les tâches plus rapidement. Pour activer la configuration basée sur AMI, omettez le bloc LifeCycleConfig de la configuration du groupe d’instances lorsque vous créez des clusters à l’aide de l’API ou, lorsque vous utilisez la console SageMaker AI, sélectionnez « Aucun » sous Scripts de cycle de vie dans Configuration personnalisée. Pour une personnalisation supplémentaire en plus de la ligne de base de configuration basée sur l’AMI, un script d’extension peut être fourni en spécifiant le paramètre OnInitComplete et Sources3Uri dans le bloc LifeCycleConfig via l’API, ou en fournissant l’URI S3 dans le champ « Fichier de script d’extension en S3 » dans Configuration personnalisée lors de l’utilisation de la console. Les scripts de configuration du cycle de vie personnalisés restent entièrement pris en charge pour les cas d’utilisation nécessitant un contrôle total du provisionnement.
La configuration du cycle de vie des nœuds basée sur l’AMI pour les clusters Slurm utilisant le provisionnement continu est disponible dans toutes les Régions AWS où SageMaker HyperPod est disponible. Pour commencer, consultez Démarrage avec SageMaker HyperPod à l’aide de l’AWS CLI ou Démarrage avec SageMaker HyperPod à l’aide de la console SageMaker AI dans le Guide de l’utilisateur de SageMaker.