谷歌TPU使用方法

Posted 2022-04-11 herosunly

谷歌TPU使用方法

• 1. 数据和框架说明
• • 1.1 读取数据
  • 1.2 创建验证集
• 2. 验证环境是否为Colab TPU
• 3. 定义模型
• 4. 不同版本TF使用TPU的方法
• • 4.1 TF 1.13
  • 4.2 TF 1.14
• 5. 评估模型

  本文主要介绍的是Colab TPU的使用方法，运行前需要先把运行时类型修改为TPU。有同学私信说本文主要是针对TF 1.14版本，但在该环境下部分函数无法使用，如fit_generator。所以希望能将解决方案具体分享下。再加上自己平时用的更多的是TF 1.13（! pip install tensorflow==1.13.1）版本。为了让大家能够熟练使用旧版本和新版本，所以特意总结下这两个版本的不同使用方法。

  最新更新时间：2019.10.17。根据谷歌官网（https://cloud.google.com/tpu/docs/supported-versions#supported-tensorflow），colab TPU最高支持tf版本为1.13。如果是自己的数据集，无法正常使用