Modèles de référence compatibles

Cloud TPU fournit un ensemble de modèles de référence optimisés pour un entraînement rapide et précis.

Cloud TPU est compatible avec les versions majeures et mineures suivantes des frameworks TensorFlow, PyTorch et JAX/FLAX.

La numérotation des versions de TensorFlow a été modifiée avec la version 2.5.0. Les principaux numéros de version de TensorFlow se terminent par 0 et tous les numéros de version de correctifs se terminent par des chiffres supérieurs à 0. Par exemple, TF 2.10.0 est une version majeure et TF 2.10.1 est une version mineure.

Pour exécuter la dernière version compatible du framework, vérifiez s'il existe des versions de correctif dans la version majeure. Si tel est le cas, vous pouvez exécuter la dernière version de correctif compatible plutôt que la version majeure.

Framework	Version majeure	Catégorie du modèle	Modèles de référence	Versions compatibles
PyTorch	2.x	Classification d'images	ResNet-PyTorch	1.13, 2.0
		JAX	La plus récente	Les grands modèles de langage	MaxText	latest
JAX/FLAX	La plus récente	Classification d'images	ResNet50	latest