通过现有 Deep Learning VM Image 实例创建新映像

在新的虚拟机实例上安装 NVIDIA 驱动程序可能非常耗时,尤其是在创建大量映像时。避免这种情况的一种方法是创建您自己的映像,该映像基于 Deep Learning VM 映像之一,但已经预先安装了 NVIDIA 驱动程序。

本主题介绍如何基于现有 Deep Learning VM 映像创建新映像。

创建新实例

首先,请按照以下主题之一中的说明创建新实例。请务必在新实例中至少包含一个 GPU。

验证 NVIDIA 驱动程序安装

实例启动后,请验证 NVIDIA 驱动程序是否已安装:

  1. 通过 SSH 连接到您的映像:
    gcloud compute ssh "DEPLOYMENT_NAME"
  2. 运行以下命令:
    nvidia-smi

如果已安装驱动程序,则可以继续操作。

停止实例

通过运行以下命令停止实例:

gcloud compute instances stop "DEPLOYMENT_NAME"

创建您自己的映像

现在,您可以基于已停止的实例创建自己的映像。在命令行运行以下命令,并为新映像提供名称和系列名称:

gcloud compute images create "NEW_IMAGE_NAME" \
        --source-disk DEPLOYMENT_NAME \
        --source-disk-zone ZONE \
        --family NEW_FAMILY_NAME

命令运行完毕后,您会获得一个预装了 NVIDIA 驱动程序的新映像,可用来创建新的 Deep Learning VM 实例。