¿Cómo utilizo una función de Lambda para crear una instancia de cuaderno de SageMaker AI?

Resolución

Para crear una función de Lambda que a su vez cree una instancia de cuaderno de SageMaker AI, sigue estos pasos:

1. Crea un rol de ejecución de Lambda de AWS Identity and Access Management (IAM) con una política adjunta que permita la acción sagemaker:CreateNotebookInstance. Por ejemplo, crea un rol que tenga adjunta la política AmazonSageMakerFullAccess que permita la acción sagemaker:CreateNotebookInstance.

2. Crea un rol de ejecución de IAM para SageMaker AI. Asegúrate de adjuntar los permisos necesarios.

3. Abre la consola de Lambda.

4. Selecciona Crear función.

5. Selecciona Crear desde cero y, a continuación, define los siguientes parámetros:
   En Nombre de la función, introduce un nombre para la función.
   En Versión ejecutable, selecciona una de las opciones de Python.
   En Cambiar el rol de ejecución predeterminado, elige Usar un rol existente.
   En Rol existente, seleccione el rol de IAM que creaste en el paso 1.

6. Selecciona Crear función.

7. En la sección Código, introduce el siguiente código:

   import osimport boto3import time
   
   INSTANCE_TYPE = os.environ['INSTANCE_TYPE']
   NOTEBOOK_NAME = os.environ['NOTEBOOK_NAME']
   ROLE=os.environ['ROLE']
   
   sagemaker = boto3.client('sagemaker')
   
   def lambda_handler(event, context):
       sagemaker_notebook = sagemaker.create_notebook_instance(
           NotebookInstanceName = NOTEBOOK_NAME +'-'+str(int(time.time())),
           InstanceType = INSTANCE_TYPE,
           RoleArn=ROLE
       )
   
   print("New Amazon SageMaker notebook instance created.")

8. Selecciona la pestaña Configuración y, a continuación, selecciona Variables de entorno.

9. Selecciona Editar.

10. Elige Agregar variable de entorno y, a continuación, crea tres variables de entorno con las siguientes opciones.
    Introduce los siguientes valores para la primera variable de entorno:
    En Clave, introduce ROL.
    En Valor, introduce el nombre de recurso de Amazon (ARN) para el rol de ejecución de SageMaker AI que creaste en el paso 2.
    Introduce los siguientes valores para la segunda variable de entorno:
    En Clave, escribe INSTANCE_TYPE.
    En Valor, introduce el tipo de instancia de Amazon Elastic Compute Cloud (Amazon EC2) para la instancia del cuaderno.
    Introduce los siguientes valores para la tercera variable de entorno:
    En Clave, escribe NOTEBOOK_NAME.
    En Valor, introduce un nombre para tu cuaderno.

11. Selecciona Guardar.

12. Selecciona la pestaña Probar.

13. En Evento de prueba, elige Crear nuevo evento o Editar evento guardado y, a continuación, elige el evento guardado que quieres usar. O bien, utiliza un evento de prueba vacío («{}»).

14. Selecciona Guardar y, a continuación, selecciona Probar.

15. Abre la consola de SageMaker AI para confirmar que se está inicializando una instancia de cuaderno.
    Nota: Si se agota el tiempo de espera de la prueba de función de Lambda, abre la función de Lambda. Elige Configuración, ve a la sección Configuración general y, a continuación, aumenta el valor del tiempo de espera. El valor predeterminado es de tres segundos.

Después de crear la función de Lambda, puedes crear un desencadenador para ejecutar automáticamente la función según el evento de otro servicio de AWS.