Recursos para usar PyTorch con Amazon SageMaker AI
Puede utilizar Amazon SageMaker AI para entrenar e implementar un modelo con código de PyTorch personalizado. Los modelos y estimadores de PyTorch del SageMaker Python SDK y el contenedor de PyTorch de código abierto de SageMaker AI facilitan la escritura de un script de PyTorch y su ejecución en SageMaker AI. La siguiente sección contiene material de referencia que puede utilizar para aprender a utilizar PyTorch con SageMaker AI.
¿Qué quiere hacer?
- Quiero entrenar un modelo de PyTorch personalizado en SageMaker AI.
-
Para ver un ejemplo de cuaderno de Jupyter, consulte los PyTorch example notebook
en el repositorio de GitHub de ejemplos de Amazon SageMaker AI. Para obtener la documentación, consulte Train a Model with PyTorch
. - Tengo un modelo de PyTorch que he entrenado en SageMaker AI y quiero implementarlo en un punto de conexión alojado.
-
Para obtener más información, consulte Deploy PyTorch models
. - Tengo un modelo de PyTorch que he entrenado fuera de SageMaker AI y quiero implementarlo en un punto de conexión de SageMaker AI.
-
Para obtener más información, consulte Deploy your own PyTorch model
. - Quiero ver la documentación de la API para las clases de PyTorch del Amazon SageMaker Python SDK
. -
Para obtener más información, consulte PyTorch Classes
. - Quiero encontrar el repositorio de contenedores PyTorch de SageMaker AI.
-
Para obtener más información, consulte el SageMaker AI PyTorch Container GitHub repository
. - Quiero obtener información sobre las versiones de PyTorch compatibles con contenedores de aprendizaje profundo de AWS.
-
Para obtener más información, consulte Available Deep Learning Container Images
.
Para obtener información general sobre la escritura de scripts de entrenamiento de PyTorch y el uso de estimadores y modelos de PyTorch con SageMaker AI, consulte Uso de PyTorch con el SDK de SageMaker Python
