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

我可以在Cloud Datalab中导入tensorflow并运行模型，但是如何配置它以使用GPU？如何配置Google Cloud Datalab以将GPU用于TensorFlow？

这里的文档仅约机会谈，我不知道如何通过Datalab配置： https://cloud.google.com/ml-engine/docs/how-tos/using-gpus

我已经试过这样：

datalab create --machine-type standard_gpu ml

，当我选择了我们 - 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'.

来源

2017-03-09 skunkwerk

您链接到（https://cloud.google.com/ml-engine/docs/how-tos/using-gpus）的页面不说明如何使用GPU的磨片n使用Google Cloud ML Engine API进行培训，您可以使用Datalab提交ML Engine API的作业。 Datalab中包含一些样本（例如样本/ ML工具箱/图像分类/花卉/服务端到端.ipynb）

如果您想在Datalab VM上本地训练Tensorflow模型，那么Datalab必须在当前不支持的Datalab虚拟机上运行GPU。

来源

2017-03-09 19:13:22

我想在云中运行它，但只是在Datalab中运行Python代码（不使用API）。那可能吗？ – skunkwerk

有两个问题：运行Datalab的虚拟机没有安装NVidia驱动程序（除非您自己安装），并且由于需要NVidia许可证，Docker镜像本身没有安装CUDA或GPU支持。此外，必须安装Tensorflow的GPU版本。 –

根据此document on GPUs on Compute Engine，只有这些区域现在提供GPU机器。

美国west1-B
美国east1-d
欧洲west1-B
亚太east1-A

可以在GUI https://console.cloud.google.com/compute/instances创建GPU实例

来源

2017-03-25 22:27:26 Raja

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

datalab beta create-gpu datalab-instance-name

来源

2017-11-14 10:52:59 Paul

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

回答

相关问题