ExecutionTemplate

筆記本執行工作負載的說明。

JSON 表示法
{
  "scaleTier": enum (ScaleTier),
  "masterType": string,
  "acceleratorConfig": {
    object (SchedulerAcceleratorConfig)
  },
  "labels": {
    string: string,
    ...
  },
  "inputNotebookFile": string,
  "containerImageUri": string,
  "outputNotebookFolder": string,
  "paramsYamlFile": string,
  "parameters": string,
  "serviceAccount": string,
  "jobType": enum (JobType),
  "kernelSpec": string,
  "tensorboard": string,

  // The following is a list of mutually exclusive fields. At most one of the
  // fields will be set in a response:
  "dataprocParameters": {
    object (DataprocParameters)
  },
  "vertexAiParameters": {
    object (VertexAIParameters)
  }
  // End of mutually exclusive fields.
}
欄位
scaleTier
(deprecated)

enum (ScaleTier)

這是必要旗標,用於執行筆記本的硬體規模層級。[已淘汰] 將停止服務。因為目前僅支援 CUSTOM。

masterType

string

指定要用於訓練工作主要工作站的虛擬機器類型。如果 scaleTier 設為 CUSTOM,就必須指定這個欄位。

您可以在這個欄位中直接使用特定 Compute Engine 機型。mapTypeId 支援的類型如下:

  • n1-standard-4
  • n1-standard-8
  • n1-standard-16
  • n1-standard-32
  • n1-standard-64
  • n1-standard-96
  • n1-highmem-2
  • n1-highmem-4
  • n1-highmem-8
  • n1-highmem-16
  • n1-highmem-32
  • n1-highmem-64
  • n1-highmem-96
  • n1-highcpu-16
  • n1-highcpu-32
  • n1-highcpu-64
  • n1-highcpu-96

或者,您也可以使用下列舊版機型:

  • standard
  • large_model
  • complex_model_s
  • complex_model_m
  • complex_model_l
  • standard_gpu
  • complex_model_m_gpu
  • complex_model_l_gpu
  • standard_p100
  • complex_model_m_p100
  • standard_v100
  • large_model_v100
  • complex_model_m_v100
  • complex_model_l_v100

最後,如要使用 TPU 進行訓練,請在這個欄位中指定 cloud_tpu。進一步瞭解使用 TPU 訓練的特殊設定選項。

acceleratorConfig

object (SchedulerAcceleratorConfig)

執行筆記本的硬體設定 (數量和加速器類型)。

labels

map (key: string, value: string)

執行的標籤。如果已排定執行時間,系統會納入「nbs-scheduled」欄位。否則會立即執行,且內含的欄位會是「nbs-immediate」。使用欄位可有效率地為各種執行類型建立索引。

包含 "key": value 組合清單的物件,範例:{ "name": "wrench", "mass": "1.3kg", "count": "3" }。

inputNotebookFile

string

要執行的筆記本檔案路徑。必須位於 Google Cloud Storage bucket 中。格式:gs://{bucket_name}/{folder}/{notebook_file_name}。範例:gs://notebook_user/scheduled_notebooks/sentiment_notebook.ipynb

containerImageUri

string

DLVM 的容器映像檔 URI。範例:'gcr.io/deeplearning-platform-release/base-cu100'。如需更多範例,請參閱:https://cloud.google.com/ai-platform/deep-learning-containers/docs/choosing-container

outputNotebookFolder

string

要寫入的筆記本資料夾路徑。必須位於 Google Cloud Storage bucket 路徑中。格式:gs://{bucket_name}/{folder}。範例:gs://notebook_user/scheduled_notebooks

paramsYamlFile

string

執行期間要在筆記本中覆寫的參數。請參閱 https://papermill.readthedocs.io/en/latest/usage-parameterize.html,瞭解如何在輸入筆記本中指定參數,並在 YAML 檔案中傳遞這些參數。例如:gs://notebook_user/scheduled_notebooks/sentiment_notebook_params.yaml

parameters

string

在「inputNotebookFile」筆記本中使用的參數。

serviceAccount

string

執行作業時要使用的服務帳戶電子郵件地址。您必須具備指定服務帳戶的 iam.serviceAccounts.actAs 權限。

jobType

enum (JobType)

這項執行作業使用的 Job 類型。

kernelSpec

string

要使用的核心規格名稱。如果執行目標上的核心規格名稱與輸入筆記本檔案中的名稱不符,就必須指定這個名稱。

tensorboard

string

這個執行作業要將 TensorBoard 記錄上傳至的 Vertex AI [TensorBoard] 資源名稱。格式:projects/{project}/locations/{location}/tensorboards/{tensorboard}

執行類型的參數。注意:目前 Vertex AI 工作沒有額外參數。以下列出互斥欄位。回應中最多會設定一個欄位:
dataprocParameters

object (DataprocParameters)

在 Dataproc JobType 執行作業中使用的參數。

vertexAiParameters

object (VertexAIParameters)

在 Vertex AI JobType 執行作業中使用的參數。

互斥欄位的結尾。

ScaleTier

這是必要旗標,指定機器類型、工作站和參數伺服器的備用資源數量。

列舉
SCALE_TIER_UNSPECIFIED 未指定規模層級。
BASIC 單一 worker 執行個體。這個層級適合學習如何使用 Cloud ML,以及使用小型資料集實驗新模型。
STANDARD_1 許多工作人員和少數參數伺服器。
PREMIUM_1 大量工作站搭配多個參數伺服器。
BASIC_GPU 單一 worker 執行個體,搭載 K80 GPU。
BASIC_TPU 具有 Cloud TPU 的單一 worker 執行個體。
CUSTOM

「自訂」層級並非固定層級,而是讓您使用自己的叢集規格。使用這個層級時,請按照下列指南設定值,以設定處理叢集:

  • 您必須設定 ExecutionTemplate.masterType,指定主節點要使用的機器類型。這是唯一需要完成的設定。

SchedulerAcceleratorConfig

硬體加速器的定義。請注意,並非所有 type 和 coreCount 的組合都有效。如要瞭解有效組合,請參閱「Compute Engine 上的 GPU」。不支援 TPU。

JSON 表示法
{
  "type": enum (SchedulerAcceleratorType),
  "coreCount": string
}
欄位
type

enum (SchedulerAcceleratorType)

這項加速器的類型。

coreCount

string (int64 format)

這個加速器的核心數量。

SchedulerAcceleratorType

AI Platform Training 作業的硬體加速器類型。

列舉
SCHEDULER_ACCELERATOR_TYPE_UNSPECIFIED 未指定加速器類型。預設為無 GPU。
NVIDIA_TESLA_K80 Nvidia Tesla K80 GPU。
NVIDIA_TESLA_P100 Nvidia Tesla P100 GPU。
NVIDIA_TESLA_V100 Nvidia Tesla V100 GPU。
NVIDIA_TESLA_P4 Nvidia Tesla P4 GPU。
NVIDIA_TESLA_T4 Nvidia Tesla T4 GPU。
NVIDIA_TESLA_A100 Nvidia Tesla A100 GPU。
TPU_V2 TPU v2。
TPU_V3 TPU v3。

JobType

這項執行作業使用的後端。

列舉
JOB_TYPE_UNSPECIFIED 未指定類型。
VERTEX_AI aiplatform.googleapis.com中的自訂工作。執行的預設值。
DATAPROC 在叢集上以 Dataproc 作業的形式執行。https://cloud.google.com/dataproc/docs/reference/rest/v1/projects.regions.jobs

DataprocParameters

在 Dataproc JobType 執行作業中使用的參數。

JSON 表示法
{
  "cluster": string
}
欄位
cluster

string

用於執行 Dataproc 執行的叢集 URI。格式:projects/{PROJECT_ID}/regions/{REGION}/clusters/{CLUSTER_NAME}

VertexAIParameters

在 Vertex AI JobType 執行作業中使用的參數。

JSON 表示法
{
  "network": string,
  "env": {
    string: string,
    ...
  }
}
欄位
network

string

應與工作對等互連的 Compute Engine 網路全名。例如,projects/12345/global/networks/myVPC。格式為 projects/{project}/global/networks/{network} 形式。其中 {project} 是專案編號,如 12345 所示,{network} 則是網路名稱。

網路必須已設定私人服務存取權。如未指定,工作就不會與任何網路對等互連。

env

map (key: string, value: string)

環境變數。最多可指定 100 個環境變數,且不得重複。範例:GCP_BUCKET=gs://my-bucket/samples/

包含 "key": value 組合清單的物件,範例:{ "name": "wrench", "mass": "1.3kg", "count": "3" }。