我可以在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'.
答案 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机器。