Google Cloud AI 平台上的 TensorFlow 2答案

【问题标题】：Tensorflow 2 on Google Cloud AI platformGoogle Cloud AI 平台上的 TensorFlow 2
【发布时间】：2020-04-02 03:26:06
【问题描述】：

Google Cloud AI 平台何时支持 Tensorflow 2 有什么消息吗？

根据列表，1.15 是最后一个支持的版本：https://cloud.google.com/ml-engine/docs/runtime-version-list

【问题讨论】：

如果需要，您可以使用自定义容器进行模型训练。 cloud.google.com/ai-platform/training/docs/…

标签： google-cloud-ml

【解决方案1】：

我们将在 2 月初正式支持 TF 2.1，因为相应的服务会有较大的变化。感谢您的耐心等待！

【讨论】：

对此有任何更新，我真的很想开始使用gpu（甚至tpu）训练，atm可以在ai平台上安装tensorflow 2.1但只能使用cpu训练
@MarkEssel 降级到 TF 2.0 以使用 GPU。
支持似乎从 3 月 9 日开始cloud.google.com/ai-platform/training/docs/runtime-version-list

【解决方案2】：

将依赖项 tensorflow-gpu==2.0 添加到 setup.py 文件以在 AI Platform 运行时版本 1.15 中使用 Tensorflow 2.0 进行训练。由于 CUDA 库不匹配，Tensorflow 2.1 无法使用 GPU，但您可以使用 CPU 进行训练。注意其他依赖不要覆盖tensorflow-gpu。免责声明：我以这种方式只培训了一份工作，这是一个非常老套的解决方案。

【讨论】：

在 ai-platform 支持带有 gpu 的 tensorflow 2.1 之前，我在 cpu 训练中使用相同的方法

【解决方案3】：

Tensorflow 2.1 运行时版本自 3 月 9 日起推出

https://cloud.google.com/ai-platform/training/docs/runtime-version-list

【讨论】：