受支持的 TensorFlow 和 Cloud TPU 版本

TensorFlow 是一个用于研究和生产的开源机器学习库。它用于在所有 Cloud TPU 版本上训练和评估模型。此页面显示了当前受支持的 TensorFlow 和 Cloud TPU 版本的列表。

受支持的 TensorFlow 版本和官方模型

下表展示了当前受支持的 TensorFlow 版本以及每个 TensorFlow 版本支持的所有官方模型。

TensorFlow 版本 支持开始日期 支持结束日期 模型类别 支持的模型
TF 2.4.0 2020 年 12 月 21 日 (结束日期尚未确定) 图片分类 ResNet-2.xMNIST-2.xEfficientNet-2.x
语言建模 Transformer-2.xBERT-2.x
对象检测 RetinaNet-2.x
图片分割 Mask-RCNN-2.xShapeMask-2.x
推荐系统 NCF-2.x
TF 2.3.1 2020 年 10 月 1 日 (结束日期尚未确定) 图片分类 ResNet-2.xMNIST-2.xEfficientNet-2.x
语言建模 Transformer-2.xBERT-2.x
对象检测 RetinaNet-2.x
图片分割 Mask-RCNN-2.xShapeMask-2.x
推荐系统 NCF-2.x
TF 2.3 2020 年 7 月 28 日 (结束日期尚未确定) 图片分类 ResNet-2.xMNIST-2.xEfficientNet-2.x
语言建模 Transformer-2.xBERT-2.x
对象检测 RetinaNet-2.x
图片分割 Mask-RCNN-2.xShapeMask-2.x
推荐系统 NCF-2.x
TF 2.2.2 2020 年 1 月 12 日 (结束日期尚未确定) 图片分类 ResNet-2.xMNIST-2.xEfficientNet-2.x
语言建模 Transformer-2.xBERT-2.x
对象检测 RetinaNet-2.x
图片分割 Mask-RCNN-2.x
TF 2.2.1 2020 年 10 月 7 日 (结束日期尚未确定) 图片分类 ResNet-2.xMNIST-2.xEfficientNet-2.x
语言建模 Transformer-2.xBERT-2.x
对象检测 RetinaNet-2.x
图片分割 Mask-RCNN-2.x
TF 2.2 2020 年 5 月 8 日 (结束日期尚未确定) 图片分类 ResNet-2.xMNIST-2.xEfficientNet-2.x
语言建模 Transformer-2.xBERT-2.x
对象检测 RetinaNet-2.x
图片分割 Mask-RCNN-2.x
TF 2.1.2 2016 年 10 月 30 日 (结束日期尚未确定) 图片分类 ResNet-2.xMNIST-2.x
语言建模 Transformer-2.xBERT-2.x
对象检测 RetinaNet-2.x
TF 2.1.1 2020 年 5 月 21 日 (结束日期尚未确定) 图片分类 ResNet-2.xMNIST-2.x
语言建模 Transformer-2.xBERT-2.x
对象检测 RetinaNet-2.x
TF 2.1 2020 年 1 月 9 日 2020 年 6 月 30 日 图片分类 ResNet-2.xMNIST-2.x
语言建模 Transformer-2.xBERT-2.x
对象检测 RetinaNet-2.x
TF 1.15.5 2020 年 1 月 14 日 (结束日期尚未确定) 图片分类 ResNetAmoebaNet
EfficientNet
MNasNetMNIST
语言建模 TransformerBERT
对象检测 RetinaNet
图片分割 Mask RCNNShapeMask
TF 1.15.4 2020 年 10 月 21 日 (结束日期尚未确定) 图片分类 ResNetAmoebaNet
EfficientNet
MNasNetMNIST
语言建模 TransformerBERT
对象检测 RetinaNet
图片分割 Mask RCNNShapeMask
TF 1.15.3 2020 年 5 月 29 日 (结束日期尚未确定) 图片分类 ResNetAmoebaNet
EfficientNet
MNasNetMNIST
语言建模 TransformerBERT
对象检测 RetinaNet
图片分割 Mask RCNNShapeMask
TF 1.15.2 2020 年 5 月 12 日 2020 年 6 月 30 日 图片分类 ResNetAmoebaNet
EfficientNet
MNasNetMNIST
语言建模 TransformerBERT
对象检测 RetinaNet
图片分割 Mask RCNNShapeMask
TF 1.15 2019 年 10 月 22 日 2020 年 6 月 30 日 图片分类 ResNetAmoebaNet
EfficientNet
MNasNetMNIST
TF 1.14 2019 年 7 月 19 日 2020 年 6 月 30 日 图片分类 ResNetAmoebaNet
EfficientNet
MNasNetMNIST
语言建模
对象检测 RetinaNetMask RCNN
TF 1.13 2019 年 3 月 11 日 2020 年 6 月 30 日 图片分类 ResNetAmoebaNet
MNasNetMNIST
语言建模
对象检测 RetinaNet
PyTorch 版本 支持开始日期 支持结束日期 模型类别 支持的模型
PyTorch 1.x 2016 年 10 月 27 日 (结束日期尚未确定) 图片分类 ResNet
语言建模 FairSeq TransformerFairSeq RoBERTa
推荐系统 DLRM

受支持的 Cloud TPU 版本

如需查看有关如何选择 Cloud TPU 版本的指南,请参阅系统架构页面上的 TPU 版本文档。下表展示了受支持的 Cloud TPU 版本。

Cloud TPU 版本 支持开始 支持结束
v2-8 2018 年 2 月 12 日 (结束日期尚未确定)
v2-32 2019 年 5 月 7 日 (结束日期尚未确定)
v2-128 2019 年 5 月 7 日 (结束日期尚未确定)
v2-256 2019 年 5 月 7 日 (结束日期尚未确定)
v2-512 2019 年 5 月 7 日 (结束日期尚未确定)
v3-8 2018 年 10 月 10 日 (结束日期尚未确定)
v3-32 2019 年 5 月 7 日 (结束日期尚未确定)
v3-128 2019 年 5 月 7 日 (结束日期尚未确定)
v3-256 2019 年 5 月 7 日 (结束日期尚未确定)
v3-512 2019 年 5 月 7 日 (结束日期尚未确定)
v3-1024 2019 年 5 月 7 日 (结束日期尚未确定)
v3-2048 2019 年 5 月 7 日 (结束日期尚未确定)