PyTorch-Einbindung

Auf dieser Seite wird die PyTorch-Einbindung von Gemini Enterprise Agent Platform erläutert. Außerdem erfahren Sie, wie Sie PyTorch auf Gemini Enterprise Agent Platform verwenden. Die PyTorch-Einbindung von Gemini Enterprise Agent Platform erleichtert das Trainieren, Bereitstellen und Orchestrieren von PyTorch-Modellen in der Produktion.

Code in Notebooks ausführen

Die Agent Platform bietet zwei Optionen zum Ausführen Ihres Codes in Notebooks: Colab Enterprise und Vertex AI Workbench. Weitere Informationen zu diesen Optionen finden Sie unter Notebooklösung auswählen.

Vordefinierte Container für das Training

Die Gemini Enterprise Agent Platform bietet vordefinierte Docker-Container-Images für das Modelltraining. Diese Container sind nach Frameworks für maschinelles Lernen und Framework-Versionen organisiert und enthalten allgemeine Abhängigkeiten, die Sie in Ihrem Trainingscode verwenden können. Informationen darüber, welche PyTorch-Versionen vordefinierte Trainingscontainern umfassen und über das Trainieren von Modellen mit einem vordefinierten Trainingscontainer finden Sie unter Vordefinierte Container für benutzerdefiniertes Training.

Vordefinierte Container für die Bereitstellung von Inferenzanfragen

Die Gemini Enterprise Agent Platform bietet vordefinierte Docker-Container-Images für die Bereitstellung von Batch- und Onlineinferenzen. Diese Container sind nach Frameworks für maschinelles Lernen und Framework-Versionen organisiert und enthalten allgemeine Abhängigkeiten, die Sie in Ihrem Inferenzcode verwenden können. Informationen darüber, welche PyTorch-Versionen vordefinierte Inferenzcontainer umfassen und über das Bereitstellen von Modellen mit einem vordefinierten Inferenzcontainer finden Sie unter Vordefinierte Container für benutzerdefiniertes Training.

Verteiltes Training

Sie können das verteilte Training von PyTorch-Modellen auf der Gemini Enterprise Agent Platform verwenden. Beim Multi-Worker-Training können Sie mit dem Reduzierungsserver die Leistung für kollektive Vorgänge noch weiter optimieren. Weitere Informationen zum verteilten Training auf der Gemini Enterprise Agent Platform finden Sie unter Verteiltes Training.

Ressourcen für die Verwendung von PyTorch auf der Gemini Enterprise Agent Platform

Weitere Informationen, darunter zur Verwendung von PyTorch in der Gemini Enterprise Agent Platform, finden Sie in den folgenden Ressourcen:

Nächste Schritte

  • Anleitung: Mit der Gemini Enterprise Agent Platform ein PyTorch-Bildklassifizierungsmodell in einer der vordefinierten Containerumgebungen der Gemini Enterprise Agent Platform mithilfe der Google Cloud Console trainieren.

    Eine detaillierte Anleitung dazu finden Sie direkt in der Google Cloud Console. Klicken Sie dazu einfach auf Anleitung:

    Anleitung