如何创建预装了张量流或 pytorch 的 GPU 虚拟机?

Question

我正在使用Google Cloud。有没有一种简单的方法可以创建带有GPU的VM，并预装TensorFlow或pytorch？

Answer 1

使用云启动器：

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

您可以在计算引擎上启动 VM，选项包括一种选择所需 GPU 硬件的方法。您必须确保区域中有 GPU 配额，但除此之外，它既快速又轻松。

您还可以使用 gcloud 从命令行启动 VM：

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"

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

Answer 2

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

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

在 Colab 中还有许多关于 GPU 使用的教程，例如，

• 谷歌合作实验室的有用片段。包括免费的图形处理器。
• 谷歌合作实验室免费GPU教程

Kaggle Kernels现在也支持GPU。有关更多信息，请参阅此处。