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.

Imágenes de Docker de SageMaker IA prediseñadas para aprendizaje profundo

Amazon SageMaker AI proporciona imágenes de Docker prediseñadas que incluyen marcos de aprendizaje profundo y otras dependencias necesarias para el entrenamiento y la inferencia. Para obtener una lista completa de las imágenes de Docker prediseñadas administradas por SageMaker IA, consulte las rutas de registro de Docker y el código de ejemplo.

Uso del SDK de Python para SageMaker IA

Con el SDK de SageMaker Python, puede entrenar e implementar modelos utilizando estos populares marcos de aprendizaje profundo. Para obtener instrucciones sobre la instalación y el uso del SDK, consulte Amazon SageMaker Python SDK. En la siguiente tabla se enumeran los marcos disponibles y las instrucciones sobre cómo usarlos con el SDK de SageMaker Python:

Ampliación de las imágenes prediseñadas de SageMaker AI Docker

Puede personalizar estos contenedores precompilados o ampliarlos según sea necesario. Con esta personalización, puede gestionar cualquier requisito funcional adicional para su algoritmo o modelo que no sea compatible con la imagen de Docker de SageMaker IA prediseñada. Para ver un ejemplo de ello, puedes ver Fine-tuning e implementar un modelo BerTopic en SageMaker IA con tus propios scripts y conjuntos de datos, ampliando los contenedores existentes. PyTorch

También puedes usar contenedores prediseñados para implementar tus modelos personalizados o modelos que hayan sido entrenados en un marco distinto SageMaker de la IA. Para obtener una descripción general del proceso, consulte Traiga sus propios MXNet TensorFlow o modelos previamente entrenados a Amazon. SageMaker En este tutorial, se explica cómo incorporar los artefactos del modelo entrenados a la SageMaker IA y alojarlos en un punto final.