Integración en PyTorch

En esta página, se explica la integración de PyTorch de Vertex AI y se proporcionan recursos que te muestran cómo usar PyTorch en Vertex AI. La integración de PyTorch de Vertex AI facilita el entrenamiento, la implementación y la organización de modelos de PyTorch en la producción.

Ejecuta código en notebooks

Vertex AI proporciona dos opciones para ejecutar tu código en notebooks, Colab Enterprise y Vertex AI Workbench. Para obtener más información sobre estas opciones, consulta Elige una solución de notebook.

Contenedores previamente compilados para el entrenamiento

Vertex AI proporciona imágenes de contenedor de Docker compiladas con anterioridad para el entrenamiento de modelos. Estos contenedores están organizados por frameworks de aprendizaje automático y versiones de framework. Además, incluyen dependencias comunes que tal vez quieras usar en tu código de entrenamiento. Para obtener información sobre qué versiones de PyTorch tienen contenedores de entrenamiento compilados previamente y cómo entrenar modelos con un contenedor de entrenamiento compilado previamente, consulta Contenedores compilados previamente para el entrenamiento personalizado.

Contenedores previamente compilados para entregar predicciones

Vertex AI proporciona imágenes de contenedor de Docker compiladas previamente para entregar predicciones por lotes y en línea. Estos contenedores están organizados por frameworks de aprendizaje automático y versiones de framework. Además, incluyen dependencias comunes que tal vez quieras usar en tu código de predicción. Para obtener información sobre qué versiones de PyTorch tienen contenedores de predicción compilados previamente y cómo entregar modelos con un contenedor de predicción compilado previamente, consulta Contenedores compilados previamente para el entrenamiento personalizado.

Entrenamiento distribuido

Puedes ejecutar el entrenamiento distribuido de modelos de PyTorch en Vertex AI. En el entrenamiento de varios trabajadores, puedes usar Reduction Server a fin de optimizar aún más el rendimiento para las operaciones colectivas de reducción total. Para obtener más información sobre el entrenamiento distribuido en Vertex AI, consulta Entrenamiento distribuido.

Recursos para usar PyTorch en Vertex AI

Para obtener más información y comenzar a usar TensorFlow en Vertex AI, consulta los siguientes recursos.

¿Qué sigue?

  • Instructivo: usa Vertex AI para entrenar un modelo de clasificación de imágenes de PyTorch en uno de los entornos de contenedores precompilados de Vertex AI mediante la consola de Google Cloud.

    Para seguir la guía paso a paso en esta tarea directamente en la consola de Google Cloud, haz clic en Guía:

    guiarme