Creación de un SageMaker HyperPod clúster con la orquestación de Amazon EKS - Amazon SageMaker AI

Las traducciones son generadas a través de traducción automática. En caso de conflicto entre la traducción y la version original de inglés, prevalecerá la version en inglés.

Creación de un SageMaker HyperPod clúster con la orquestación de Amazon EKS

El siguiente tutorial muestra cómo crear un SageMaker HyperPod clúster nuevo y configurarlo con Amazon EKS Orchestration a través de la interfaz de usuario de la consola SageMaker AI.

Cree un clúster

Para ir a la página de SageMaker HyperPod clústeres y elegir Amazon EKS Orchestration, siga estos pasos.

  1. Abre la consola Amazon SageMaker AI en https://console.aws.amazon.com/sagemaker/.

  2. Seleccione HyperPod Clusters en el panel de navegación izquierdo y, a continuación, Cluster Management.

  3. En la página SageMaker HyperPod Clústeres, elija Crear HyperPod clúster.

  4. En el menú desplegable Crear HyperPod clúster, elija Orchestrated by Amazon EKS.

  5. En la página de creación de clústeres de EKS, verá dos opciones. Elija la opción que mejor se adapte a sus necesidades.

    1. Configuración rápida: para comenzar de inmediato con la configuración predeterminada, seleccione Configuración rápida. Con esta opción, la SageMaker IA creará nuevos recursos, como VPC, subredes, grupos de seguridad, un bucket de Amazon S3, una función de IAM y FSx para Lustre, durante el proceso de creación del clúster.

    2. Configuración personalizada: para integrarse con los recursos de AWS existentes o tener requisitos específicos de red, seguridad o almacenamiento, seleccione Configuración personalizada. Con esta opción, puede optar por utilizar los recursos existentes o crear unos nuevos. Además, puede personalizar la configuración que mejor se adapte a sus necesidades.

En la sección Configuración rápida, siga estos pasos para crear su HyperPod clúster con la orquestación de Amazon EKS.

Configuración general

Especifique un nombre para el nuevo clúster. Después de crear el clúster no puede cambiarle el nombre.

Grupos de instancias

Para añadir un grupo de instancias, elija Agregar grupo. Cada grupo de instancias se puede configurar de una forma diferente y se puede crear un clúster heterogéneo que conste de varios grupos de instancias con diversos tipos de instancias. Para implementar un clúster, debe agregar al menos un grupo de instancias. Siga estos pasos para agregar un grupo de instancias.

  1. Para Tipo del grupo de instancias, seleccione Estándar o Grupo de instancias restringido (RIG). En general, se selecciona Estándar pues proporciona un entorno computacional de uso general sin restricciones de seguridad adicionales. Grupo de instancias restringido (RIG) es un entorno especializado para la personalización de modelos fundacionales, como Amazon Nova. Para obtener más información sobre la configuración de RIG para la personalización de modelos de Amazon Nova, consulte Personalización de Amazon Nova en Amazon SageMaker HyperPod.

  2. En Nombre, especifique un nombre para el grupo de instancias.

  3. En Capacidad de instancia, elija la capacidad bajo demanda o un plan de entrenamiento para reservar recursos de computación.

  4. En Tipo de instancia, elija la instancia del grupo de instancias.

    importante

    Asegúrese de elegir un tipo de instancia con bastantes cuotas y que tenga suficientes direcciones IP sin asignar para su cuenta. Para ver o solicitar cuotas adicionales, consulte SageMaker HyperPod cuotas.

  5. En Cantidad de instancias, especifique un número entero que no supere la cuota de instancias de uso del clúster. Para este tutorial, introduzca 1 para los tres grupos.

  6. En Zona de disponibilidad de destino, elija la zona de disponibilidad en la que se aprovisionarán las instancias. La zona de disponibilidad debe corresponder a la ubicación de la capacidad de computación acelerada.

  7. En Volumen de almacenamiento adicional por instancia (GB) - opcional, especifique un número entero entre 1 y 16 384 para establecer el tamaño de un volumen adicional de Elastic Block Store (EBS) en gigabytes (GB). El volumen de EBS está asociado a cada instancia del grupo de instancias. La ruta de montaje predeterminada para el volumen de EBS adicional es /opt/sagemaker. Una vez que el clúster se haya creado correctamente, puede utilizar SSH en las instancias del clúster (nodos) y verificar si el volumen de EBS está montado correctamente ejecutando el comando df -h. La conexión de un volumen de EBS adicional proporciona un almacenamiento estable, fuera de la instancia y persistente de forma independiente, tal y como se describe en la sección Amazon EBS volumes de la Guía del usuario de Amazon Elastic Block Store.

  8. En Comprobaciones exhaustivas del estado de las instancias, elija la opción deseada. Las comprobaciones exhaustivas de estado supervisan el estado de las instancias durante la creación y después de las actualizaciones de software, y recuperan automáticamente las instancias con error mediante reinicios o sustituciones cuando están habilitadas.

  9. Si su tipo de instancia admite la partición de GPU con GPU de instancias múltiples (MIG), puede habilitar la configuración de particiones de GPU para el grupo de instancias. La partición de la GPU te permite GPUs dividirla en particiones más pequeñas y aisladas para mejorar el uso de los recursos. Para obtener más información, consulte Uso de particiones de GPU en Amazon SageMaker HyperPod.

    1. Activa la opción Usar partición de GPU para habilitar la partición de GPU para este grupo de instancias.

    2. Selecciona un perfil de partición de GPU entre las opciones disponibles para tu tipo de instancia. Cada perfil define la configuración del segmento de GPU y la asignación de memoria.

  10. Elija Agregar grupo de instancias.

Valores predeterminados de la configuración rápida

En esta sección se enumeran todos los ajustes predeterminados para la creación del clúster, incluidos todos los AWS recursos nuevos que se crearán durante el proceso de creación del clúster. Revise la configuración predeterminada.

En la sección Configuración personalizada, siga estos pasos para crear su primer HyperPod clúster con la orquestación de Amazon EKS.

Configuración general

Especifique un nombre para el nuevo clúster. Después de crear el clúster no puede cambiarle el nombre.

En Recuperación de instancias, elija Automática - recomendado o Ninguna.

Red

Configure los ajustes de red dentro del clúster y in-and-out del clúster. Para la orquestación del SageMaker HyperPod clúster con Amazon EKS, la VPC se establece automáticamente en la que se configuró con el clúster de EKS que seleccionó.

  1. En el caso de la VPC, elige tu propia VPC si ya tienes una que dé acceso de SageMaker IA a tu VPC. Para crear una VPC nueva, siga las instrucciones de la sección Creación de una VPC de la Guía del usuario de Amazon Virtual Private Cloud. Puedes dejarlo como Ninguno para usar la VPC de SageMaker IA predeterminada.

  2. Para el bloque IPv4 CIDR de VPC, introduzca la IP de inicio de la VPC.

  3. En el caso de las zonas de disponibilidad, elija las zonas de disponibilidad (AZ) en las que se HyperPod crearán las subredes para el clúster. Elija una AZs que coincida con la ubicación de su capacidad informática acelerada.

  4. En Grupos de seguridad, elija grupos de seguridad que estén asociados al clúster de Amazon EKS o cuyo tráfico entrante esté permitido por el grupo de seguridad asociado al clúster de Amazon EKS. Para crear nuevos grupos de seguridad, vaya a la consola de Amazon VPC.

Orquestación

Siga estos pasos para crear o seleccionar un clúster de Amazon EKS para usarlo como orquestador.

  1. En Clúster de EKS, elija crear un nuevo clúster de Amazon EKS o utilizar uno existente.

    Si debe crear un nuevo clúster de EKS, puede crearlo en la sección Clúster de EKS sin abrir la consola de Amazon EKS.

    nota

    La subred de VPC que elija debe ser HyperPod privada.

    Tras enviar una solicitud de creación de un nuevo clúster de EKS, espere a que el clúster de EKS pase a estar Active.

  2. En Versión de Kubernetes, elija una versión en el menú desplegable. Para obtener más información sobre las versiones de Kubernetes, consulte Descripción del ciclo de vida de las versiones de Kubernetes en EKS en la Guía del usuario de Amazon EKS.

  3. En Operadores, elija Usa gráficos y complementos predeterminados de Helm o No instales operadores. La opción predeterminada es Usar gráficos y complementos predeterminados de Helm, que se utilizará para instalar los operadores en el clúster de EKS. Para obtener más información sobre los gráficos y complementos predeterminados de Helm, consulte el helm_chart GitHubrepositorio. Para obtener más información, consulte Instalación de paquetes en el clúster de Amazon EKS mediante Helm.

  4. En Operadores habilitados, consulte la lista de operadores habilitados. Para editar los operadores, desmarque la casilla situada en la parte superior y elija los operadores que desee activar en el clúster de EKS.

    nota

    Para utilizarlos HyperPod con EKS, debe instalar los gráficos y complementos de Helm que habiliten a los operadores en el clúster de EKS. Estos componentes configuran EKS como el plano de control HyperPod y proporcionan la configuración necesaria para la gestión y organización de la carga de trabajo.

Grupos de instancias

Para añadir un grupo de instancias, elija Agregar grupo. Cada grupo de instancias se puede configurar de una forma diferente y se puede crear un clúster heterogéneo que conste de varios grupos de instancias con diversos tipos de instancias. Para implementar un clúster, debe agregar al menos un grupo de instancias. Siga estos pasos para agregar un grupo de instancias.

  1. Para Tipo del grupo de instancias, seleccione Estándar o Grupo de instancias restringido (RIG). En general, se selecciona Estándar pues proporciona un entorno computacional de uso general sin restricciones de seguridad adicionales. Grupo de instancias restringido (RIG) es un entorno especializado para la personalización de modelos fundacionales, como Amazon Nova. Para obtener más información sobre la configuración de RIG para la personalización de modelos de Amazon Nova, consulte Personalización de Amazon Nova en Amazon SageMaker HyperPod.

  2. En Nombre, especifique un nombre para el grupo de instancias.

  3. En Capacidad de instancia, elija la capacidad bajo demanda o un plan de entrenamiento para reservar recursos de computación.

  4. En Tipo de instancia, elija la instancia del grupo de instancias.

    importante

    Asegúrese de elegir un tipo de instancia con bastantes cuotas y que tenga suficientes direcciones IP sin asignar para su cuenta. Para ver o solicitar cuotas adicionales, consulte SageMaker HyperPod cuotas.

  5. En Cantidad de instancias, especifique un número entero que no supere la cuota de instancias de uso del clúster. Para este tutorial, introduzca 1 para los tres grupos.

  6. En Zona de disponibilidad de destino, elija la zona de disponibilidad en la que se aprovisionarán las instancias. La zona de disponibilidad debe corresponder a la ubicación de la capacidad de computación acelerada.

  7. En Volumen de almacenamiento adicional por instancia (GB) - opcional, especifique un número entero entre 1 y 16 384 para establecer el tamaño de un volumen adicional de Elastic Block Store (EBS) en gigabytes (GB). El volumen de EBS está asociado a cada instancia del grupo de instancias. La ruta de montaje predeterminada para el volumen de EBS adicional es /opt/sagemaker. Una vez que el clúster se haya creado correctamente, puede utilizar SSH en las instancias del clúster (nodos) y verificar si el volumen de EBS está montado correctamente ejecutando el comando df -h. La conexión de un volumen de EBS adicional proporciona un almacenamiento estable, fuera de la instancia y persistente de forma independiente, tal y como se describe en la sección Amazon EBS volumes de la Guía del usuario de Amazon Elastic Block Store.

  8. En Comprobaciones exhaustivas del estado de las instancias, elija la opción deseada. Las comprobaciones exhaustivas de estado supervisan el estado de las instancias durante la creación y después de las actualizaciones de software, y recuperan automáticamente las instancias con error mediante reinicios o sustituciones cuando están habilitadas. Para obtener más información, consulte Comprobaciones de estado exhaustivas

  9. Para usar la partición de GPU (opcional), si tu tipo de instancia admite la partición de GPU con una GPU de varias instancias (MIG), puedes habilitar esta opción para configurar el perfil de partición de GPU para el grupo de instancias. La partición de la GPU te permite GPUs dividirla en particiones más pequeñas y aisladas para mejorar el uso de los recursos. Para obtener más información, consulte Uso de particiones de GPU en Amazon SageMaker HyperPod.

    1. Activa la opción Usar partición de GPU para habilitar la partición de GPU para este grupo de instancias.

    2. Selecciona un perfil de partición de GPU entre las opciones disponibles para tu tipo de instancia. Cada perfil define la configuración del segmento de GPU y la asignación de memoria.

  10. Elija Agregar grupo de instancias.

Scripts de ciclo de vida

Puede optar por utilizar los scripts de ciclo de vida predeterminados o los scripts de ciclo de vida personalizados, que se almacenarán en su bucket de Amazon S3. Puede ver los scripts de ciclo de vida predeterminados en el GitHub repositorio de Awesome Distributed Training. Para obtener más información sobre los scripts de ciclo de vida, consulte Personalización de SageMaker HyperPod clústeres mediante scripts de ciclo de vida.

  1. En Scripts de ciclo de vida, elija usar scripts de ciclo de vida predeterminados o personalizados.

  2. En bucket de S3 para scripts de ciclo de vida, elija crear un nuevo bucket o utilizar un bucket existente para almacenar los scripts de ciclo de vida.

Permisos

Elija o cree un rol de IAM que le permita HyperPod ejecutar y acceder a AWS los recursos necesarios en su nombre. Para obtener más información, consulte Función de IAM para SageMaker HyperPod.

Almacenamiento

Configure el sistema FSx de archivos de Lustre para que se aprovisione en el clúster. HyperPod

  1. Para el sistema de archivos, elija un sistema de archivos existente FSx para Lustre, cree uno nuevo FSx para Lustre o no aprovisione uno FSx para Lustre.

  2. En Rendimiento por unidad de almacenamiento, elija el rendimiento que estará disponible por TiB de almacenamiento aprovisionado.

  3. En Capacidad de almacenamiento, introduzca un valor de capacidad en TB.

  4. En el tipo de compresión de datos, seleccione habilitar LZ4la compresión de datos.

  5. En Versión Lustre, consulte el valor recomendado para los nuevos sistemas de archivos.

Etiquetas: opcional

En el caso de las etiquetas (opcional), añada pares de claves y valores al nuevo clúster y administre el clúster como un AWS recurso. Para obtener más información, consulte Tagging your AWS resources.

Implementación de recursos

Después de completar las configuraciones del clúster desde Configuración rápida o Configuración personalizada, elija la siguiente opción para iniciar el aprovisionamiento de recursos y la creación del clúster.

  • Enviar: SageMaker AI empezará a aprovisionar los recursos de configuración predeterminados y a crear el clúster.

  • Descargar los parámetros de la CloudFormation plantilla: descargará el archivo JSON de los parámetros de configuración y ejecutará el AWS CLI comando para implementar la CloudFormation pila, aprovisionar los recursos de configuración y crear el clúster. Si es necesario, puede editar el archivo JSON de parámetros descargado. Si elige esta opción, puede ver más instrucciones en Crear SageMaker HyperPod clústeres mediante CloudFormation plantillas.