View a markdown version of this page

Uso de PCS-ready DLAMI con AWS 2 PIEZAS - AWS PIEZAS
Qué incluyeBusque el PCS-ready DLAMI actualReciba notificaciones de las nuevas versionesÚselo con la infraestructura como códigoActualiza a una nueva versión

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.

Uso de PCS-ready DLAMI con AWS 2 PIEZAS

La AMI de GPU base de AWS PCS-ready DLAMI (Ubuntu 24.04) es una imagen de máquina de AWS-maintained Amazon para ejecutar cargas de trabajo HPC AI/ML y en ejecución en PC. AWS Proporciona una base lista para la producción, de modo que puede implementar clústeres en cuestión de minutos en lugar de crear y validar AMI personalizadas.

Qué incluye

PCS-ready DLAMI se basa en la AMI de GPU de Deep Learning Base (Ubuntu 24.04) y agrega los siguientes componentes de PCS: AWS

  • PCS Agent: el agente de administración de clústeres de AWS PCS

  • Slurm para AWS PCS: vienen preinstaladas varias versiones compatibles de Slurm. La versión correcta se activa automáticamente durante el lanzamiento de la instancia en función de la configuración del clúster.

  • Utilidades EFS: para montar sistemas de archivos Amazon EFS

El DLAMI de origen proporciona el sistema operativo (Ubuntu 24.04), los controladores de GPU de NVIDIA, el kit de herramientas CUDA, los controladores EFA, el cliente Lustre y otra infraestructura básica. Para obtener más información sobre estos componentes, consulte las notas de la versión de la AMI de aprendizaje profundo. https://docs.aws.amazon.com/dlami/latest/devguide/appendix-ami-release-notes.html

PCS-ready DLAMI está disponible para las arquitecturas x86_64 y arm64.

nota

PCS-ready DLAMI no incluye software de aplicaciones como AI/ML marcos (PyTorch,, JAX) TensorFlow, compiladores o bibliotecas matemáticas. Puede agregar su capa de aplicación en sistemas de archivos compartidos o creando una AMI personalizada sobre DLAMI. PCS-ready

El campo de descripción de cada AMI resume su contenido, incluido el DLAMI de origen en el que se basa, la versión del agente de PCS, las versiones de Slurm compatibles y la versión de las utilidades de EFS. Puede ver este campo en la consola de Amazon EC2 o mediante la API. describe-images El siguiente es un ejemplo de un valor de campo de descripción:

PCS-Ready DLAMI based on Deep Learning Base OSS Nvidia Driver GPU AMI (Ubuntu 24.04) 20260522. PCS Agent: 1.4.0-1. Slurm: 24.11.7-1, 25.05.7-1, 25.11.2-1. EFS Utils: 2.4.2

Busque el PCS-ready DLAMI actual

Consola de administración de AWS
Para buscar el PCS-ready DLAMI en la consola

  1. Abra la consola AWS PCS y navegue para crear o editar un grupo de nodos de procesamiento.

  2. En la sección de selección de AMI, seleccione PCS-ready AMI.

  3. Aparece un menú desplegable que muestra las PCS-ready DLAMI disponibles filtradas según la arquitectura de tipo de instancia seleccionada.

  4. Elija la AMI base de AWS PCS-ready DLAMI (Ubuntu 24.04). El menú desplegable muestra el ID de la AMI y el nombre completo de la AMI a modo de referencia.

AWS CLI

Puede recuperar el ID de AMI de PCS-ready DLAMI más reciente mediante el almacén de parámetros de Amazon EC2 Systems Manager. Sustitúyalo por region-code su. Región de AWS

  • x86_64

    aws ssm get-parameter --region region-code \
  --name /aws/service/pcs/ami/dlami-base-ubuntu2404/x86_64/latest/ami-id \
  --query "Parameter.Value" --output text

  • arm64 

    aws ssm get-parameter --region region-code \
  --name /aws/service/pcs/ami/dlami-base-ubuntu2404/arm64/latest/ami-id \
  --query "Parameter.Value" --output text

Como alternativa, puede buscar PCS-ready DLAMI por patrón de nombre:

  • x86_64

    aws ec2 describe-images --region region-code --owners amazon \
  --filters 'Name=name,Values=aws-pcs-ready-dlami-base-ubuntu2404-x86_64-*' \
            'Name=state,Values=available' \
  --query 'sort_by(Images, &CreationDate)[-1].[Name,ImageId]' --output text

  • arm64 

    aws ec2 describe-images --region region-code --owners amazon \
  --filters 'Name=name,Values=aws-pcs-ready-dlami-base-ubuntu2404-arm64-*' \
            'Name=state,Values=available' \
  --query 'sort_by(Images, &CreationDate)[-1].[Name,ImageId]' --output text

Usa el ID de AMI al crear o actualizar un grupo de nodos de procesamiento.

Reciba notificaciones de las nuevas versiones

Puede recibir notificaciones cada vez que se publique un nuevo PCS-ready DLAMI. Las notificaciones se publican en Amazon SNS siguiendo el siguiente tema:

arn:aws:sns:us-west-2:265886551188:pcs-ready-dlami-release-notifications

Los mensajes sobre este tema se publican cuando se publica un nuevo PCS-ready DLAMI.

Para suscribirse a las notificaciones de publicación de PCS-ready DLAMI

  1. Abra la consola de Amazon SNS.

  2. En la barra de navegación, cámbiala Región de AWS a US West (Oregón), si es necesario. Debe seleccionar la región en la que se creó la notificación de SNS a la que se está suscribiendo.

  3. En el panel de navegación, elige Suscripciones y, a continuación, elige Crear suscripción.

  4. En el cuadro de diálogo Crear suscripción, haga lo siguiente:

    1. En el tema ARN, introduzca lo siguiente: arn:aws:sns:us-west-2:265886551188:pcs-ready-dlami-release-notifications

    2. En Protocolo, elige Correo electrónico.

    3. Para Endpoint, introduzca la dirección de correo electrónico en la que desea recibir las notificaciones.

    4. Seleccione Crear subscripción.

  5. Recibirá un correo electrónico de confirmación con el asunto Notificación de AWS: confirmación de suscripción. Abra el correo electrónico y elija Confirmar suscripción para completar la suscripción.

Úselo con la infraestructura como código

La ruta del parámetro SSM proporciona una referencia estable que siempre se resuelve en el ID de AMI más reciente. Puede utilizarla en CloudFormation las plantillas para seleccionar automáticamente las nuevas versiones al volver a implementarlas:

AmiId: '{{resolve:ssm:/aws/service/pcs/ami/dlami-base-ubuntu2404/x86_64/latest/ami-id}}'

Actualiza a una nueva versión

AWS publica las versiones actualizadas de PCS-ready DLAMI cuando se actualiza la AMI de la GPU de la base de aprendizaje profundo de origen o cuando se actualizan los componentes de PCS (PCS Agent o Slurm para PCS). Para actualizar el clúster, recupere el ID de AMI más reciente mediante la búsqueda de nombres o parámetros SSM descrita anteriormente y, a continuación, actualice cada grupo de nodos de procesamiento para que haga referencia al nuevo ID de AMI.