Marcos y AWS regiones compatibles - Amazon SageMaker

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.

Marcos y AWS regiones compatibles

Antes de usar el cargador de datos de filtrado SageMaker inteligente, compruebe si el marco que ha elegido es compatible, si los tipos de instancia están disponibles en su AWS cuenta y si su AWS cuenta se encuentra en una de las AWS regiones compatibles.

nota

SageMaker El filtrado inteligente permite el entrenamiento de PyTorch modelos con el paralelismo de datos tradicional y el paralelismo de datos distribuidos, lo que permite crear réplicas de modelos en todos los trabajadores y utilizar la operación. GPU AllReduce No funciona con las técnicas de paralelismo de modelos, incluido el paralelismo de datos fragmentados. Como el tamizado SageMaker inteligente funciona para trabajos de paralelismo de datos, asegúrate de que el modelo que entrenes quepa en cada memoria. GPU

Marcos admitidos

SageMaker smart sifting es compatible con los siguientes marcos de aprendizaje profundo y está disponible a través de AWS Deep Learning Containers.

Temas

PyTorch

Marcos Versión de marco Contenedor de aprendizaje profundo URI
PyTorch 2.1.0

763104351884.dkr.ecr.region.amazonaws.com/pytorch-training:2.1.0-gpu-py310-cu121-ubuntu20.04-sagemaker

Para obtener más información sobre los contenedores prediseñados, consulte SageMaker Framework Containers en el GitHub repositorio de AWS Deep Learning Containers.

Regiones de AWS

Los contenedores empaquetados con la biblioteca de SageMaker filtrado inteligente están disponibles en el Regiones de AWS lugar donde se encuentran en servicio los AWS Deep Learning Containers.

Tipos de instancias

Puede utilizar el tamizado SageMaker inteligente para cualquier trabajo de PyTorch formación en cualquier tipo de instancia. Le recomendamos que utilice instancias P4d, P4de o P5.