TPU定制芯片可与Google Cloud ML一起使用

时间:2016-10-14 20:29:33

标签: tensorflow google-cloud-ml google-cloud-tpu

使用TensorFlow时,哪种类型的硬件用作Google Cloud ML的一部分?只有CPU或张量处理单元(自定义卡)可用吗?

参见此article

3 个答案:

答案 0 :(得分:2)

Cloud ML目前专注于CPU。 GPU和TPU将来可用。

答案 1 :(得分:0)

在撰写本文时(2017年12月),可以使用GPU,请参阅https://cloud.google.com/ml-engine/docs/training-overview

如果您使用gcloud命令行实用程序,则可以使用在向mlengine提交作业时添加--scale-tier BASIC_GPU选项。目前,这可以在特斯拉K80上运行您的张量流代码。

还有一个CUSTOM比例级别,允许更复杂的配置并允许访问P100 GPU(参见https://cloud.google.com/ml-engine/docs/using-gpus)。

根据https://cloud.google.com/tpu/,TPU服务处于“alpha”状态,需要注册才能了解详情。

答案 2 :(得分:0)