如何使用带有 GPU 的计算引擎创建数据实验室?
How to create datalab with a compute engine with a GPU?
- 我可以创建具有特定 CE 实例类型的数据实验室,但无法将 GPU 添加到该实例。
- 我可以使用 GPU 创建 CE 实例,但不知道如何在其上 运行 datalab。
请帮忙。
编辑:
罗斯上面的回答是正确的。现在存在 'datalab beta create-gpu' 命令。
原答案(过时):
Datalab 目前不支持使用 GPU。这是一个 datalab github issue #1275.
Datalab 运行 所在的 VM 没有安装 NVidia 驱动程序(除非您自己安装),并且 Datalab Docker 映像本身没有 CUDA 或 GPU由于需要 NVidia 许可证而安装了支持。
用datalab create tool just uses "gcloud compute instances create" calls to make the VMs, and you could theoretically copy those commands and convert them to "gcloud beta compute instances create" with "--accelerator" options, but you'd still have to create your own version of the Datalab docker image to run on it with the NVidia drivers installed (which requires accepting the NVidia license agreement). And that may also require running it using nvidia_docker代替正常的docker。
另请参阅:
Datalab 现在支持 GPU。请在此处查看 datalab beta create-gpu
上的文档:
https://cloud.google.com/datalab/docs/quickstart#create_and_connect_to_a_cloud_datalab_instance
- 我可以创建具有特定 CE 实例类型的数据实验室,但无法将 GPU 添加到该实例。
- 我可以使用 GPU 创建 CE 实例,但不知道如何在其上 运行 datalab。 请帮忙。
编辑: 罗斯上面的回答是正确的。现在存在 'datalab beta create-gpu' 命令。
原答案(过时): Datalab 目前不支持使用 GPU。这是一个 datalab github issue #1275.
Datalab 运行 所在的 VM 没有安装 NVidia 驱动程序(除非您自己安装),并且 Datalab Docker 映像本身没有 CUDA 或 GPU由于需要 NVidia 许可证而安装了支持。
用datalab create tool just uses "gcloud compute instances create" calls to make the VMs, and you could theoretically copy those commands and convert them to "gcloud beta compute instances create" with "--accelerator" options, but you'd still have to create your own version of the Datalab docker image to run on it with the NVidia drivers installed (which requires accepting the NVidia license agreement). And that may also require running it using nvidia_docker代替正常的docker。
另请参阅:
Datalab 现在支持 GPU。请在此处查看 datalab beta create-gpu
上的文档:
https://cloud.google.com/datalab/docs/quickstart#create_and_connect_to_a_cloud_datalab_instance