如何配置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？

回答

相關問題