创建运行时模板

本页介绍了如何在 Colab Enterprise 中创建运行时模板。

如需在笔记本中运行代码,请使用名为运行时的计算资源。您可以使用默认运行时或根据运行时模板创建的运行时。通过创建运行时模板,您可以配置模板以根据需要优化运行时的性能、费用和其他特性。

详细了解运行时和运行时模板

须知事项

  1. Sign in to your Google Cloud account. If you're new to Google Cloud, create an account to evaluate how our products perform in real-world scenarios. New customers also get $300 in free credits to run, test, and deploy workloads.
  2. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

    Go to project selector

  3. Make sure that billing is enabled for your Google Cloud project.

  4. Enable the Vertex AI, Dataform, and Compute Engine APIs.

    Enable the APIs

  5. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

    Go to project selector

  6. Make sure that billing is enabled for your Google Cloud project.

  7. Enable the Vertex AI, Dataform, and Compute Engine APIs.

    Enable the APIs

所需的角色

如需确保您的用户账号具有在 Colab Enterprise 中创建运行时模板所需的权限,请让您的管理员向您的用户账号授予项目的 Colab Enterprise Admin (roles/aiplatform.colabEnterpriseAdmin) IAM 角色。如需详细了解如何授予角色,请参阅管理对项目、文件夹和组织的访问权限

您的管理员还可以通过自定义角色或其他预定义角色向您的用户账号授予所需的权限。

创建运行时模板

如需创建运行时模板,您可以使用 Google Cloud 控制台或 Google Cloud CLI。

控制台

如需创建运行时模板,请执行以下操作:

  1. 在 Google Cloud 控制台中,前往 Colab Enterprise 运行时模板页面。

    进入运行时模板

  2. 点击  新建模板

    此时会显示创建新的运行时模板对话框。

运行时基础知识

  1. 运行时基本信息部分中,输入显示名称

  2. 区域菜单中,选择需要运行时模板的区域。

  3. 可选:添加运行时模板的说明

  4. 可选:如需添加标签,请点击  添加标签,然后输入对。 如需添加更多标签,请重复此步骤。

  5. 点击继续

配置计算

  1. 配置计算部分的机器类型菜单中,选择机器类型。如需了解机器类型,请参阅机器系列资源和比较指南

    如果您选择了包含 GPU 的机器类型,请选择加速器类型加速器数量。 如果您无法选择所需的 GPU 数量,则可能需要增加配额。请参阅提交配额增加申请

  2. 磁盘类型菜单中,选择磁盘类型。

  3. 磁盘大小字段中,输入大小(以 GB 为单位)。

  4. 空闲机器关停部分中:

    • 如需关闭空闲机器关停,请清除启用空闲机器关停

    • 如需更改非活跃时间段,请在关停前保持非活跃状态的时间(分钟)部分中,将数字更改为所需的非活跃分钟数。在 Google Cloud 控制台中,您可以将此设置设为 10 到 1440 之间的任何整数值。

  5. 点击继续

网络和安全

  1. 网络和安全部分的网络菜单中,选择一个网络。如果您未选择网络,则系统会选择您的默认网络。

  2. 子网菜单中,选择一个子网。

  3. 如需停用公共互联网访问权限,请清除启用公共互联网访问权限复选框。

  4. 如需停用最终用户凭据访问权限,请清除启用最终用户凭据复选框。

完成运行时模板的创建

点击创建以完成运行时模板的创建。

您的运行时模板会显示在运行时模板标签页上的列表中。

gcloud

在使用下面的命令数据之前,请先进行以下替换:

  • DISPLAY_NAME:运行时模板的显示名称。
  • PROJECT_ID:您的项目 ID。
  • REGION:要将运行时模板部署到的区域。
  • MACHINE_TYPE:要用于运行时的机器类型
  • ACCELERATOR_TYPE:要为运行时使用的硬件加速器类型。
  • ACCELERATOR_COUNT:要为运行时使用的加速器数量。

执行以下命令:

Linux、macOS 或 Cloud Shell

gcloud colab runtime-templates create --display-name="DISPLAY_NAME" \
    --project=PROJECT_ID \
    --region=REGION \
    --machine-type=MACHINE_TYPE \
    --accelerator-type=ACCELERATOR_TYPE \
    --accelerator-count=ACCELERATOR_COUNT

Windows (PowerShell)

gcloud colab runtime-templates create --display-name="DISPLAY_NAME" `
    --project=PROJECT_ID `
    --region=REGION `
    --machine-type=MACHINE_TYPE `
    --accelerator-type=ACCELERATOR_TYPE `
    --accelerator-count=ACCELERATOR_COUNT

Windows (cmd.exe)

gcloud colab runtime-templates create --display-name="DISPLAY_NAME" ^
    --project=PROJECT_ID ^
    --region=REGION ^
    --machine-type=MACHINE_TYPE ^
    --accelerator-type=ACCELERATOR_TYPE ^
    --accelerator-count=ACCELERATOR_COUNT

如需详细了解用于从命令行创建运行时模板的命令,请参阅 gcloud CLI 文档

授予对运行时模板的访问权限

创建运行时模板后,您必须授予该模板的访问权限,以使主账号可以使用它。主账号只有在具有以下权限时才能根据运行时模板创建运行时

  • 运行时模板的访问权限。
  • 创建运行时所需的权限。

请参阅管理对运行时模板的访问权限

删除运行时模板

如需删除运行时模板,请执行以下操作:

  1. 在 Google Cloud 控制台中,前往 Colab Enterprise 运行时模板页面。

    进入运行时模板

  2. 区域菜单中,选择包含运行时模板的区域。

  3. 选择要删除的运行时模板。

  4. 点击 删除

  5. 点击确认

问题排查

本部分介绍了如何解决在 Colab Enterprise 中创建运行时模板时遇到的问题。

无法选择足够的 GPU

创建运行时模板时,您无法选择所需的 GPU 数量。这可能是因为您没有足够的配额。

Colab Enterprise 为 GPU 使用 Compute Engine 配额。如需了解详情,请参阅 Compute Engine 配额和限制概览

如需解决此问题,请申请更高的配额限制

后续步骤