支持的 TPU 版本
如需确定训练模型的最有效 TPU 硬件版本，请参阅系统架构页面上的 TPU 硬件版本文档。如需查看 Cloud TPU 支持的参考模型列表，请参阅支持的模型页面。
下表展示了支持的 Cloud TPU 版本。
| 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 日 | （结束日期尚未确定） |
TPU 软件版本
您应使用的 TPU 软件版本取决于 TPU 架构、TPU 虚拟机或 TPU 节点，以及您使用的机器学习框架（TensorFlow、PyTorch 或 JAX）。
TPU 虚拟机
TensorFlow
使用与编写模型的 TensorFlow 版本相匹配的 TPU 软件版本。例如，如果您使用的是 TensorFlow 2.8.0，请使用 tpu-vm-tf-2.8.0 TPU 软件版本。如果您使用的是 TPU Pod，请使用 tpu-vm-tf-2.8.0-pod。目前适用于单设备 TPU 的 TensorFlow TPU 虚拟机软件版本有：
- TPU-vm-tf-2.6.2
- TPU-vm-tf-2.7.1
- TPU-vm-tf-2.8.0
如需详细了解 TensorFlow 补丁程序版本，请参阅支持的 TensorFlow 补丁程序版本。
如需指定 TPU 虚拟机 Pod，请将 -pod 添加到单设备列表中显示的版本，例如 tpu-vm-tf-2.8.0-pod。
当您创建 TPU 虚拟机时，TensorFlow 虚拟机上已预安装最新版本的 TensorFlow。
TPU 虚拟机在创建时预装了 TensorFlow 和相应的 LibTPU 库。如果您要创建自己的虚拟机映像，请指定以下 TensorFlow TPU 软件版本和对应的 LibTPU 版本：
| TensorFlow 版本 | libTPU.so 版本 |
|---|---|
| 2.8.* | 1.2.0 |
| 2.7.* | 1.1.1 |
| 2.6.* | 1.0.0 |
PyTorch
使用与编写模型的 PyTorch 版本相匹配的 TPU 软件版本。例如，如果您使用的是 PyTorch 1.10，请使用 tpu-vm-pt-1.10 TPU 软件版本。如果您使用的是 TPU Pod，请使用 tpu-vm-pt-1.10-pod。当前支持的 TPU 软件版本如下：
- TPU-vm-pt-1.8 和 tpu-vm-pt-1.8-pod
- TPU-vm-pt-1.9 和 tpu-vm-pt-1.9-pod
- TPU-vm-pt-1.10 和 tpu-vm-pt-1.10-pod
创建 TPU 虚拟机时，TPU 虚拟机上会预安装最新版本的 PyTorch。
如需更改当前的 PyTorch 软件版本，请参阅更改 PyTorch 版本。
JAX
您必须在 TPU 虚拟机上手动安装 JAX，因此不存在特定于 JAX 的 TPU 软件版本。您可以使用 TensorFlow 列出的任何软件版本。
TPU 节点
TensorFlow
使用与编写模型的 TensorFlow 版本相匹配的 TPU 运行时版本。例如，如果您使用的是 TF 2.6.0，请使用 2.6.0 TPU 软件版本。TensorFlow 专用 TPU 软件版本如下：
- 1.15.5
- 2.1.4
- 2.2.3
- 2.3.4
- 2.4.3
- 2.5.3
- 2.6.2
- 2.7.1
- Nightly
如需详细了解 TensorFlow 补丁程序版本，请参阅支持的 TensorFlow 补丁程序版本。
创建 TPU 节点时，TPU 节点上会预安装最新版本的 TensorFlow。
PyTorch
使用与编写模型的 PyTorch 版本相匹配的 TPU 运行时版本。例如，如果您使用的是 PyTorch 1.9，请使用 pytorch-1.9 运行时版本。
PyTorch 专用的 TPU 软件版本如下所示：
- pytorch-1.6
- pytorch-1.7
- pytorch-1.8
- pytorch-1.9
- pytorch-1.10
- pytorch-1.11
- pytorch-nightly
创建 TPU 节点时，TPU 节点上会预安装最新版本的 PyTorch。
JAX
您必须在 TPU 虚拟机上手动安装 JAX，因此没有预安装特定于 JAX 的 TPU 软件版本。您可以使用 TensorFlow 列出的任何软件版本。