在 Colab Enterprise 中建立執行階段範本

本頁面說明如何在 Colab Enterprise 建立執行階段範本。

如要在筆記本中執行程式碼，請使用名為「執行階段」的運算資源。您可以使用預設執行階段，或透過執行階段範本建立執行階段。建立執行階段範本後，您可以根據需求設定範本，將執行階段的效能、成本和其他特性最佳化。

進一步瞭解執行階段和執行階段範本。

事前準備

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

    Roles required to select or create a project

    • Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
    • Create a project: To create a project, you need the Project Creator role (roles/resourcemanager.projectCreator), which contains the resourcemanager.projects.create permission. Learn how to grant roles.

    Go to project selector

  2. Verify that billing is enabled for your Google Cloud project.

  3. Enable the Agent Platform, Dataform, and Compute Engine APIs, if any are not already enabled.

    Roles required to enable APIs

    To enable APIs, you need the serviceusage.services.enable permission. If you created the project, then you likely already have this permission through the Owner role (roles/owner). Otherwise, you can get this permission through the Service Usage Admin role (roles/serviceusage.serviceUsageAdmin). Learn how to grant roles.

    Enable the APIs

必要的角色

如要取得在 Colab Enterprise 建立執行階段範本所需的權限，請要求管理員授予您專案的 Colab Enterprise 管理員 (roles/aiplatform.colabEnterpriseAdmin) IAM 角色。如要進一步瞭解如何授予角色，請參閱「管理專案、資料夾和組織的存取權」。

您或許也能透過自訂角色或其他預先定義的角色，取得必要權限。

建立執行階段範本

如要建立執行階段範本，可以使用 Google Cloud 控制台、Google Cloud CLI、REST API 或 Terraform。

控制台

如要建立執行階段範本，請按照下列步驟操作：

  1. 前往 Google Cloud 控制台的 Colab Enterprise「Runtime templates」(執行階段範本) 頁面。

    前往「Runtime templates」(執行階段範本) 頁面

  2. 按一下「 New template」(新增範本)。

    「Create new runtime template」(建立新的執行階段範本) 對話方塊隨即顯示。

執行階段基本資訊

  1. 在「Runtime basics」(執行階段基本資訊) 部分，輸入「Display name」(顯示名稱)。

  2. 在「Region」(區域) 選單中，選取執行階段範本所在的區域。

  3. 選用：新增執行階段範本的說明。

  4. 選用：如要新增標籤，請按一下 「Add label」(新增標籤)，然後輸入鍵和值組合。如要新增更多標籤，請重複這個步驟。

  5. 按一下「繼續」。

設定運算資源

  1. 在「Configure compute」(設定運算) 區段的「Machine type」(機器類型) 選單中，選取機器類型。如要瞭解機器類型，請參閱機器家族資源與比較指南。

    如果選取含有 GPU 的機型，請選取「Accelerator type」(加速器類型)和「Accelerator count」(加速器數量)。如果無法選取所需的 GPU 數量，可能需要提高配額。請參閱「要求調整配額」。

  2. 在「Data disk type」(資料磁碟類型) 選單中，選取磁碟類型。

  3. 在「Data disk size」(資料磁碟大小) 欄位中，輸入大小 (以 GB 為單位)。

  4. 在「Idle shutdown」(閒置關機) 部分：

    • 如要關閉閒置關機功能，請清除「Enable idle shutdown」(啟用閒置關機功能)。

    • 如要變更閒置時間長度，請在「關機前的閒置時間 (分鐘)」中，將數字變更為所需的閒置分鐘數。在 Google Cloud 控制台中，這項設定可設為 10 到 1440 之間的任何整數值。

  5. 按一下「繼續」。

環境

  1. 在「Environment」(環境) 部分，選取「Environment」(環境)。 預設值為「Latest」(最新) (目前為 Python 3.13)。

  2. 選用：在「Post-startup script」(執行開機指令碼後) 欄位中，輸入執行開機指令碼後的 URI。如要進一步瞭解如何使用啟動後指令碼，請參閱「使用啟動後指令碼」。

  3. 選用：在執行開機指令碼後網址下方，選取執行開機指令碼後的行為。預設行為是「Run once」(執行一次)。詳情請參閱「開機指令碼行為」。

  4. 選用：如要新增環境變數，請按一下 「Add env variable」(新增環境變數)，然後輸入「Key」(鍵) 和「Value」(值) 配對。如要新增更多環境變數，請重複這個步驟。

  5. 按一下「繼續」。

網路與安全

  1. 在「Networking and security」(網路和安全性) 區段中，從「Network」(網路) 選單選取網路。如未選取網路，系統會選取預設網路。

  2. 在「Subnetwork」(子網路) 選單中選取子網路。

  3. 如要關閉公開網際網路存取權，請清除「Enable public internet access」(啟用公開網際網路存取權)。

  4. 如要關閉使用者憑證存取權，請清除「啟用使用者憑證」。

完成建立執行階段範本

按一下「Create」(建立)，完成建立執行階段範本。

執行階段範本會顯示在「Runtime templates」(執行階段範本) 分頁的清單中。

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 說明文件。

REST

使用任何要求資料之前，請先修改下列項目的值：

  • REGION：執行階段範本所在的區域。
  • PROJECT_ID：您的專案 ID。
  • DISPLAY_NAME：執行階段範本的顯示名稱。
  • MACHINE_TYPE：執行階段要使用的機器類型。
  • ACCELERATOR_TYPE：執行階段要使用的硬體加速器類型。
  • ACCELERATOR_COUNT：執行階段要使用的加速器數量。

HTTP 方法和網址：

POST https://REGION-aiplatform.googleapis.com/v1/projects/PROJECT_ID/locations/REGION/notebookRuntimeTemplates

JSON 要求內文：

{
  "displayName": "DISPLAY_NAME",
  "machineSpec": {
    {
      "machineType": MACHINE_TYPE
      "acceleratorType": ACCELERATOR_TYPE,
      "acceleratorCount": ACCELERATOR_COUNT,
    }
  },
}

如要傳送要求，請選擇以下其中一個選項：

curl

將要求主體儲存在名為 request.json 的檔案中，然後執行下列指令：

curl -X POST \
     -H "Authorization: Bearer $(gcloud auth print-access-token)" \
     -H "Content-Type: application/json; charset=utf-8" \
     -d @request.json \
     "https://REGION-aiplatform.googleapis.com/v1/projects/PROJECT_ID/locations/REGION/notebookRuntimeTemplates"

PowerShell

將要求主體儲存在名為 request.json 的檔案中，然後執行下列指令：

$cred = gcloud auth print-access-token
$headers = @{ "Authorization" = "Bearer $cred" }

Invoke-WebRequest `
    -Method POST `
    -Headers $headers `
    -ContentType: "application/json; charset=utf-8" `
    -InFile request.json `
    -Uri "https://REGION-aiplatform.googleapis.com/v1/projects/PROJECT_ID/locations/REGION/notebookRuntimeTemplates" | Select-Object -Expand Content
如果成功，回應主體會包含 Operation 的執行個體。

詳情請參閱 notebookRuntimeTemplates.create REST API 說明文件。

授予執行階段範本存取權

建立執行階段範本後，您必須授予存取權，主體才能使用範本。主體必須具備下列條件，才能從執行階段範本建立執行階段：

  • 可存取執行階段範本。
  • 建立執行階段所需的權限。

後續步驟