En esta página, se explica la integración de PyTorch de Gemini Enterprise Agent Platform y se proporcionan recursos que te muestran cómo usar PyTorch en Gemini Enterprise Agent Platform. La integración de PyTorch de Gemini Enterprise Agent Platform facilita el entrenamiento, la implementación y la organización de modelos de PyTorch en la producción.
Ejecuta código en notebooks
Agent Platform 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
Gemini Enterprise Agent Platform 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 compilados previamente para entregar inferencias
Gemini Enterprise Agent Platform proporciona imágenes de contenedor de Docker compiladas con anterioridad para entregar inferencias en línea y por lotes. 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 inferencia. Para obtener información sobre qué versiones de PyTorch tienen contenedores de inferencia compilados previamente y cómo entregar modelos con un contenedor de inferencia compilado previamente, consulta Contenedores compilados previamente para el entrenamiento personalizado.
Entrenamiento distribuido
Puedes ejecutar el entrenamiento distribuido de modelos de PyTorch en Gemini Enterprise Agent Platform. 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 Gemini Enterprise Agent Platform, consulta Entrenamiento distribuido.
Recursos para usar PyTorch en Gemini Enterprise Agent Platform
Para obtener más información y comenzar a usar PyTorch en Gemini Enterprise Agent Platform, consulta los siguientes recursos.
- Cómo entrenar y ajustar modelos de PyTorch en Gemini Enterprise Agent Platform: Obtén información para usar el entrenamiento de Gemini Enterprise Agent Platform para compilar y entrenar un modelo de clasificación de textos de opiniones con PyTorch y el ajuste de hiperparámetros de Gemini Enterprise Agent Platform para ajustar los hiperparámetros de los modelos de PyTorch.
- Cómo implementar modelos de PyTorch en Gemini Enterprise Agent Platform: Explora la implementación de un modelo de Pytorch mediante TorchServe como un contenedor personalizado. Para ello, implementa los artefactos de modelo en un servicio de Vertex AI Inference.
- Organiza los flujos de trabajo de PyTorch ML en canalizaciones de Agent Platform: Consulta cómo compilar y organizar canalizaciones de AA para entrenar e implementar modelos de PyTorch en Google Cloud Gemini Enterprise Agent Platform con canalizaciones de Agent Platform.
- Flujos de trabajo de AA escalables con PyTorch en Kubeflow Pipelines y Vertex Pipelines: Consulta ejemplos de flujos de trabajo de AA basados en PyTorch en OSS Kubeflow Pipelines (parte del proyecto de Kubeflow) y canalizaciones de Agent Platform. Compartimos nuevos componentes integrados de PyTorch agregados a Kubeflow Pipelines.
- Entrega modelos de imágenes de PyTorch con contenedores precompilados en Agent Platform: Este notebook implementa un modelo de clasificación de imágenes de PyTorch en Agent Platform con imágenes de entrega de PyTorch compiladas con anterioridad.
¿Qué sigue?
- Instructivo: Usa Gemini Enterprise Agent Platform para entrenar un modelo de clasificación de imágenes de PyTorch en uno de los entornos de contenedores precompilados de Gemini Enterprise Agent Platform con la Google Cloud consola.
Para seguir la guía paso a paso sobre esta tarea directamente en la Google Cloud consola, haz clic en Guiarme: