Preparación y carga de los scripts de ciclo de vida - Amazon SageMaker AI
Preparación de los scripts de ciclo de vidaCreación del archivo de configuraciónVerificación de los archivos en el bucket de Amazon S3

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.

Preparación y carga de los scripts de ciclo de vida

Tras crear todos los recursos necesarios, tendrá que configurar los scripts del ciclo de vida del clúster. SageMaker HyperPod Estos scripts de ciclo de vida proporcionan una configuración básica que puede usar para crear un clúster de HyperPod Slurm básico.

Preparación de los scripts de ciclo de vida

Siga estos pasos para obtener los scripts de ciclo de vida.

  1. Descargue los scripts del ciclo de vida del GitHub repositorio a su máquina.

  2. Cargue los scripts del ciclo de vida en el bucket de Amazon S3 en Aprovisionamiento de recursos básicos con el comando de la CLI cp.

    aws s3 cp --recursive LifeCycleScripts/base-config s3://${ROOT_BUCKET_NAME}/LifeCycleScripts/base-config

Creación del archivo de configuración

Siga estos pasos para crear el archivo de configuración y cargarlo en el mismo bucket de Amazon S3 en el que almacena los scripts de ciclo de vida.

  1. Cree un archivo de configuración denominado provisioning_parameters.json con el siguiente contenido. Tenga en cuenta que slurm_sns_arn es opcional. Si no se proporciona, no HyperPod configurará las notificaciones de Amazon SNS.

    cat <<EOF > /tmp/provisioning_parameters.json
{
  "version": "1.0.0",
  "workload_manager": "slurm",
  "controller_group": "$CONTOLLER_IG_NAME",
  "login_group": "my-login-group",
  "worker_groups": [
    {
      "instance_group_name": "$COMPUTE_IG_NAME",
      "partition_name": "dev"
    }
  ],
  "fsx_dns_name": "$SLURM_FSX_DNS_NAME",
  "fsx_mountname": "$SLURM_FSX_MOUNT_NAME",
  "slurm_configurations": {
    "slurm_database_secret_arn": "$SLURM_DB_SECRET_ARN",
    "slurm_database_endpoint": "$SLURM_DB_ENDPOINT_ADDRESS",
    "slurm_shared_directory": "/fsx",
    "slurm_database_user": "$DB_USER_NAME",
    "slurm_sns_arn": "$SLURM_SNS_FAILOVER_TOPIC_ARN"
  }
}
EOF

  2. Cargue el archivo provisioning_parameters.json en el mismo bucket de Amazon S3 en el que se almacenan los scripts de ciclo de vida.

    aws s3 cp /tmp/provisioning_parameters.json s3://${ROOT_BUCKET_NAME}/LifeCycleScripts/base-config/provisioning_parameters.json

Verificación de los archivos en el bucket de Amazon S3

Tras cargar todos los scripts de ciclo de vida y el archivo provisioning_parameters.json, el bucket de Amazon S3 debería tener el siguiente aspecto.

Imagen que muestra todos los scripts de ciclo de vida cargados en el bucket de Amazon S3 en la consola de Amazon Simple Storage Service.

Para obtener más información, consulte Comience con los scripts de ciclo de vida básicos proporcionados por HyperPod.