如何使用带有GPU的计算引擎创建datalab？

Question

1. 我可以使用特定的CE实例类型创建datalab，但是无法将GPU添加到该实例。
2. 我可以使用GPU创建CE实例，但后来不知道如何在其上运行datalab。 请帮忙。

Answer 1

编辑： 罗斯的答案是正确的。现在存在'datalab beta create-gpu'命令。

原始答案（过时）： Datalab目前不支持使用GPU。这是datalab github issue #1275。

运行Datalab的VM没有安装NVidia驱动程序（除非您自己安装），并且由于需要NVidia许可证，Datalab Docker映像本身没有安装CUDA或GPU支持。< / p>

datalab create tool只使用“gcloud compute instances create”调用来制作虚拟机，理论上你可以复制这些命令并将它们转换为“gcloud beta compute instances create”和“--accelerator”选项，但是您仍然必须创建自己的Datalab docker镜像版本，并在安装了NVidia驱动程序的情况下运行（这需要接受NVidia许可协议）。这可能还需要使用nvidia_docker而不是普通的docker来运行它。

另请参阅：https://stackoverflow.com/a/42703626/5441818

Answer 2

Datalab现在支持GPU。请参阅datalab beta create-gpu此处的文档：

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