Nesta página, explicamos a integração do PyTorch da plataforma de agentes do Gemini Enterprise e oferecemos recursos que mostram como usar o PyTorch nessa plataforma. A integração do PyTorch da plataforma de agentes do Gemini Enterprise facilita o treinamento, a implantação e a orquestração de modelos do PyTorch na produção.
Executar código em notebooks
A plataforma de agentes oferece duas opções para executar o código em notebooks: Colab Enterprise e Vertex AI Workbench. Para saber mais sobre essas opções, consulte Escolher uma solução de notebook.
Contêineres pré-criados para treinamento
A plataforma de agente do Gemini Enterprise oferece imagens de contêiner do Docker pré-criadas para treinamento de modelo. Esses contêineres são organizados por frameworks de aprendizado de máquina, versões de framework e incluem dependências comuns que podem ser usadas no código de treinamento. Para saber quais versões do PyTorch têm contêineres de treinamento pré-criados e como treinar modelos com um contêiner de treinamento pré-criado, consulte Contêineres pré-criados para treinamento personalizado.
Contêineres pré-criados para veicular inferências
A plataforma de agente do Gemini Enterprise fornece imagens de contêiner do Docker pré-criadas para veicular inferências em lote e on-line. Esses contêineres são organizados por frameworks de aprendizado de máquina, versões de framework e incluem dependências comuns que podem ser usadas no código de inferência. Para saber quais versões do PyTorch têm contêineres de inferência pré-criados e como disponibilizar modelos com um contêiner de inferência pré-criado, consulte Contêineres pré-criados para treinamento personalizado.
Treinamento distribuído
É possível executar o treinamento distribuído de modelos PyTorch na plataforma de agentes do Gemini Enterprise. Para o treinamento de vários workers, é possível usar o Servidor de redução para otimizar ainda mais o desempenho para operações coletivas com redução de tudo. Para saber mais sobre o treinamento distribuído na plataforma de agentes do Gemini Enterprise, consulte Treinamento distribuído.
Recursos para usar o PyTorch na plataforma de agentes do Gemini Enterprise
Para saber mais e começar a usar o PyTorch na plataforma de agentes do Gemini Enterprise, consulte os recursos a seguir:
- Como treinar e ajustar modelos do PyTorch na plataforma de agentes do Gemini Enterprise: como usar o Treinamento da plataforma de agentes do Gemini Enterprise para criar e treinar um modelo de classificação de texto de sentimento usando o PyTorch e Ajuste de hiperparâmetros da plataforma de agentes do Gemini Enterprise para ajustar os hiperparâmetros de modelos do PyTorch.
- Como implantar modelos do PyTorch na plataforma Gemini Enterprise Agent: percorra a implantação de um modelo do Pytorch usando o TorchServe como um contêiner personalizado, implantando os artefatos de modelo para um serviço da Vertex AI Inference.
- Orquestração dos fluxos de trabalho de ML do PyTorch nos pipelines da plataforma de agentes do Gemini Enterprise: como criar e orquestrar pipelines de ML para treinamento e implantação de modelos do PyTorch na Google Cloud plataforma de agentes do Gemini Enterprise usando os pipelines da plataforma de agentes do Gemini Enterprise.
- Fluxos de trabalho de ML escalonáveis usando o PyTorch nos Kubeflow Pipelines e no Vertex Pipelines: confira exemplos de fluxos de trabalho de ML baseados em PyTorch nos Kubeflow Pipelines do OSS, (parte do projeto Kubeflow) e Pipelines da plataforma de agentes do Gemini Enterprise. Compartilhamos os novos componentes integrados do PyTorch adicionados aos Pipelines do Kubeflow.
- Como disponibilizar modelos de imagem do PyTorch com contêineres predefinidos na plataforma de agentes: este notebook implanta um modelo de classificação de imagens do PyTorch na plataforma de agentes usando imagens predefinidas do PyTorch.
A seguir
- Tutorial: use a plataforma de agentes do Gemini Enterprise para treinar um modelo de classificação
de imagens PyTorch em um dos ambientes de contêiner pré-criados da plataforma de agentes do Gemini Enterprise
usando o console Google Cloud .
Para seguir as instruções detalhadas desta tarefa diretamente no console do Google Cloud , clique em Orientação: