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 |