Google Cloud AI 平台上的 Tensorflow 2



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

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

由于服务上的重大相应更改,我们将在 2 月初正式支持 TF 2.1。感谢您的耐心等待!

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

Tensorflow 2.1运行时版本将于 3 月 9 日推出

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

最新更新