PyTorch-Einbindung

Auf dieser Seite wird die PyTorch-Einbindung der Gemini Enterprise Agent Platform erläutert. Außerdem erfahren Sie, wie Sie PyTorch auf der Gemini Enterprise Agent Platform verwenden. Die PyTorch-Einbindung der 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 Trainings container 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 Online-Inferenzanfragen. 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 Inferenz Container 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 auf der Gemini Enterprise Agent Platform, finden Sie in folgenden Ressourcen:

Nächste Schritte

  • Tutorial: Gemini Enterprise Agent Platform zum Trainieren eines PyTorch-Bildklassifizierungsmodells in einer der vordefinierten Containerumgebungen der Gemini Enterprise Agent Platform mithilfe der Google Cloud Console verwenden.

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

    Anleitung