如何创建预装了 tensorflow 或 pytorch 的 GPU VM？ [关闭]答案

【问题标题】：How to create a GPU VM with tensorflow or pytorch pre-installed? [closed]如何创建预装了 tensorflow 或 pytorch 的 GPU VM？ [关闭]
【发布时间】：2018-07-03 20:35:09
【问题描述】：

我正在使用谷歌云。有没有一种简单的方法来创建一个带有 GPU 并预装了 TensorFlow 或 pytorch 的 VM？

【问题讨论】：

标签： google-dl-platform

【解决方案1】：

使用云启动器：

https://console.cloud.google.com/launcher/details/click-to-deploy-images/deeplearning

您可以在 Compute Engine 上启动虚拟机，选项包括选择您想要的 GPU 硬件的方法。您必须确保该区域中有 GPU 配额，但除此之外，它快速且轻松。

您也可以使用 gcloud 从命令行启动虚拟机：

export IMAGE_FAMILY="tf-latest-cu92" # or put any required
export ZONE="us-west1-b"
export INSTANCE_NAME="my-instance"

gcloud compute instances create $INSTANCE_NAME \
   --zone=$ZONE  \
  --image-family=$IMAGE_FAMILY  \
  --image-project=deeplearning-platform-release  \
  --maintenance-policy=TERMINATE  \
  --accelerator='type=nvidia-tesla-v100,count=1' \
   --metadata='install-nvidia-driver=True'

对于 pytorch，请改用：

export IMAGE_FAMILY="pytorch-latest-cu91"

官方文档here，包括所有可用图像的列表。

【讨论】：

【解决方案2】：

Colaboratory 在 GCE 上运行，支持 GPU，无需设置。

网站上的 GPU 说明： https://colab.research.google.com/notebooks/gpu.ipynb

Colab 中也有很多关于 GPU 使用的教程，例如，

如今，Kaggle 内核也支持 GPU。 See here for more info.

【讨论】：