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