使用 TensorFlow 时,将哪种类型的硬件用作 Google Cloud ML 的一部分?只有 CPU 或张量处理单元(自定义卡)也可用?
参考这篇文章
使用 TensorFlow 时,将哪种类型的硬件用作 Google Cloud ML 的一部分?只有 CPU 或张量处理单元(自定义卡)也可用?
参考这篇文章
Cloud ML 目前专注于 CPU。GPU 和 TPU 将在未来推出。
在撰写本文时(2017 年 12 月),GPU 可用,请参阅https://cloud.google.com/ml-engine/docs/training-overview
如果您使用gcloud
命令行实用程序,您可以--scale-tier BASIC_GPU
在向 mlengine 提交作业时添加该选项。这目前可以在 Tesla K80 上运行您的 tensorflow 代码。
还有一个CUSTOM
规模层,它允许更复杂的配置并允许访问 P100 GPU(请参阅https://cloud.google.com/ml-engine/docs/using-gpus)。
根据https://cloud.google.com/tpu/,TPU服务处于“alpha”状态,需要注册以了解更多信息。
Cloud TPU 自 2018 年 6 月 27 日起向公众开放: https ://cloud.google.com/tpu/docs/release-notes
这是在 Google Next '18 上宣布的: