Unterstützte Referenzmodelle

Cloud TPU bietet eine Reihe von Referenzmodellen, die für schnelles und genaues Training optimiert sind.

Cloud TPU unterstützt die folgenden Haupt- und Nebenversionen der Frameworks TensorFlow, PyTorch und JAX/FLAX.

Die TensorFlow-Versionsnummerierung hat sich ab Version 2.5.0 geändert. Hauptversionsnummern von TensorFlow enden mit 0 und alle Patchversionsnummern mit Ziffern größer als 0. Beispiel: TF 2.10.0 ist eine Hauptversion und TF 2.10.1 ist eine Nebenversion.

Prüfen Sie, ob die Patch-Releases des Hauptversions-Release ausgeführt werden, um die neueste unterstützte Version auszuführen. Wenn ja, können Sie die aktuelle Patchversion anstelle der Hauptversion ausführen.

Framework Hauptversion Modellkategorie Referenzmodelle Unterstützte Versionen
PyTorch 2.x Bildklassifizierung ResNet-PyTorch 1.13, 2.0
JAX Neueste Large Language Models MaxText Neu
JAX/FLAX Neueste Klassifizierung von Bildern ResNet50 Neu