如何配置Google Cloud Datalab以将GPU用于TensorFlow?

时间:2017-03-09 19:04:08

标签: google-cloud-platform google-cloud-datalab

我可以在Cloud Datalab中导入tensorflow和运行模型,但是如何配置它以使用GPU?

此处的文档仅讨论机器,我不确定如何通过Datalab进行配置: https://cloud.google.com/ml-engine/docs/how-tos/using-gpus

我试过这个:

datalab create --machine-type standard_gpu ml

当我选择us-east1区域时,我收到错误:

Creating the instance ml
ERROR: (gcloud.compute.instances.create) Some requests did not succeed:
- Invalid value for field 'resource.machineType': 'https://www.googleapis.com/compute/v1/projects/project-160204/zones/us-east1-d/machineTypes/standard_gpu'. Machine type with name 'standard_gpu' does not exist in zone 'us-east1-d'.

3 个答案:

答案 0 :(得分:4)

现在可以使用GPU创建datalab实例: https://cloud.google.com/datalab/docs/reference/command-line/create

datalab beta create-gpu datalab-instance-name

答案 1 :(得分:3)

您链接到的页面(https://cloud.google.com/ml-engine/docs/how-tos/using-gpus)确实描述了在使用Google Cloud ML Engine API进行培训时如何使用GPU,您可以使用Datalab针对ML Engine API提交作业。其中的一些样本包含在Datalab中(例如样本/ ML工具箱/图像分类/花/服务端到端.ipnb)

如果您想在Datalab VM上本地训练Tensorflow模型,那么Datalab必须针对Datalab VM上的GPU运行,目前不支持。

答案 2 :(得分:1)

根据此document on GPUs on Compute Engine,目前只有这些区域提供GPU机器。

  • 我们-west1-B
  • 我们-east1-d
  • 欧洲-west1-B
  • 亚east1-A

您可以在gui https://console.cloud.google.com/compute/instances

中创建GPU实例