0

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

4

2 回答 2

1

使用云启动器：

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

您可以在 Compute Engine 上启动 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"

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

于 2018-07-03T20:35:09.653 回答
-1

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

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

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

如今，Kaggle 内核也支持 GPU。请参阅此处了解更多信息。

于 2018-07-04T15:58:24.083 回答