支持的参考模型

Cloud TPU 提供一组经过优化、可实现快速准确训练的参考模型。

Cloud TPU 支持 TensorFlow、PyTorch 和 JAX/FLAX 的以下主要版本和次要版本。

TensorFlow 版本号在 2.5.0 版中发生了变化。主要的 TensorFlow 版本号以“0”结尾，所有补丁程序版本号以大于“0”结尾。例如，TF 2.10.0 是一个主要版本，TF 2.10.1 是一个次要版本。

如需运行最新支持的框架版本，请查看主要版本是否存在任何补丁版本。如果存在，则可以运行最新支持的补丁程序版本，而不是主要版本。

框架 主要版本 模型类别 参考模型 支持的版本
PyTorch 2.x 图片分类 ResNet-PyTorch 1.13、2.0
JAX 最新 大语言模型 MaxText latest
JAX/FLAX 最新 图片分类 ResNet50 latest