我可以在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'.
我想在雲中運行它,但只是在Datalab中運行Python代碼(不使用API)。那可能嗎? – skunkwerk
有兩個問題:運行Datalab的虛擬機沒有安裝NVidia驅動程序(除非您自己安裝),並且由於需要NVidia許可證,Docker鏡像本身沒有安裝CUDA或GPU支持。 此外,必須安裝Tensorflow的GPU版本。 –