筆記本執行工作負載的說明。
| JSON 表示法 |
|---|
{ "scaleTier": enum ( |
| 欄位 | |
|---|---|
scaleTier |
這是必要旗標,用於執行筆記本的硬體規模層級。[已淘汰] 將停止服務。因為目前僅支援 CUSTOM。 |
masterType |
指定要用於訓練工作主要工作站的虛擬機器類型。如果 您可以在這個欄位中直接使用特定 Compute Engine 機型。mapTypeId 支援的類型如下:
或者,您也可以使用下列舊版機型:
最後,如要使用 TPU 進行訓練,請在這個欄位中指定 |
acceleratorConfig |
執行筆記本的硬體設定 (數量和加速器類型)。 |
labels |
執行的標籤。如果已排定執行時間,系統會納入「nbs-scheduled」欄位。否則會立即執行,且內含的欄位會是「nbs-immediate」。使用欄位可有效率地為各種執行類型建立索引。 包含 |
inputNotebookFile |
要執行的筆記本檔案路徑。必須位於 Google Cloud Storage bucket 中。格式: |
containerImageUri |
DLVM 的容器映像檔 URI。範例:'gcr.io/deeplearning-platform-release/base-cu100'。如需更多範例,請參閱:https://cloud.google.com/ai-platform/deep-learning-containers/docs/choosing-container |
outputNotebookFolder |
要寫入的筆記本資料夾路徑。必須位於 Google Cloud Storage bucket 路徑中。格式: |
paramsYamlFile |
執行期間要在筆記本中覆寫的參數。請參閱 https://papermill.readthedocs.io/en/latest/usage-parameterize.html,瞭解如何在輸入筆記本中指定參數,並在 YAML 檔案中傳遞這些參數。例如: |
parameters |
在「inputNotebookFile」筆記本中使用的參數。 |
serviceAccount |
執行作業時要使用的服務帳戶電子郵件地址。您必須具備指定服務帳戶的 |
jobType |
這項執行作業使用的 Job 類型。 |
kernelSpec |
要使用的核心規格名稱。如果執行目標上的核心規格名稱與輸入筆記本檔案中的名稱不符,就必須指定這個名稱。 |
tensorboard |
這個執行作業要將 TensorBoard 記錄上傳至的 Vertex AI [TensorBoard] 資源名稱。格式: |
| 執行類型的參數。注意:目前 Vertex AI 工作沒有額外參數。以下列出互斥欄位。回應中最多會設定一個欄位: | |
dataprocParameters |
在 Dataproc JobType 執行作業中使用的參數。 |
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 |
「自訂」層級並非固定層級,而是讓您使用自己的叢集規格。使用這個層級時,請按照下列指南設定值,以設定處理叢集:
|
SchedulerAcceleratorConfig
硬體加速器的定義。請注意,並非所有 type 和 coreCount 的組合都有效。如要瞭解有效組合,請參閱「Compute Engine 上的 GPU」。不支援 TPU。
| JSON 表示法 |
|---|
{
"type": enum ( |
| 欄位 | |
|---|---|
type |
這項加速器的類型。 |
coreCount |
這個加速器的核心數量。 |
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 |
用於執行 Dataproc 執行的叢集 URI。格式: |
VertexAIParameters
在 Vertex AI JobType 執行作業中使用的參數。
| JSON 表示法 |
|---|
{ "network": string, "env": { string: string, ... } } |
| 欄位 | |
|---|---|
network |
應與工作對等互連的 Compute Engine 網路全名。例如, 網路必須已設定私人服務存取權。如未指定,工作就不會與任何網路對等互連。 |
env |
環境變數。最多可指定 100 個環境變數,且不得重複。範例: 包含 |