如何使用GPU創建帶有計算引擎的datalab？

Question

1. 我可以創建具有特定CE實例類型的datalab，但無法將GPU添加到該實例。
2. 我可以用GPU創建CE實例，但不知道如何在其上運行datalab。 請幫忙。

Answer 1

Datalab現在支持的GPU。請參閱datalab beta create-gpu的文檔：

https://cloud.google.com/datalab/docs/quickstart#create_and_connect_to_a_cloud_datalab_instance

Answer 2

編輯： 羅斯上面的答案是正確的。現在有一個'datalab beta create-gpu'命令。

原始答案（過時）： Datalab目前不支持使用GPU。這是一個datalab github issue #1275。

運行Datalab的虛擬機沒有安裝NVidia驅動程序（除非您自己安裝），並且由於需要NVidia許可證，Datalab Docker鏡像本身沒有安裝CUDA或GPU支持。

的datalab create tool只是使用「gcloud計算實例創建」要求，使虛擬機，並且理論上可以複製這些命令，並將其轉換爲「gcloud公測計算實例創建」與「--accelerator」選項，但你仍然必須創建自己的Datalab docker鏡像版本，以便在安裝了NVidia驅動程序（需要接受NVidia許可協議）的情況下運行該鏡像。而且這可能還需要使用nvidia_docker而不是普通的泊塢窗來運行它。

參見：https://stackoverflow.com/a/42703626/5441818