本文档介绍了如何创建和运行使用图形处理单元 (GPU) 的作业。如需详细了解 GPU 的功能和限制，请参阅 Compute Engine 文档中的 GPU 简介。
创建 Batch 作业时，您可以选择使用 GPU 来加速特定的工作负载。使用 GPU 的作业的常见应用场景包括密集型数据处理和人工智能 (AI) 工作负载（例如机器学习 (ML)）。
准备工作
- 如果您之前未使用过 Batch，请查看开始使用 Batch，并完成项目和用户的前提条件以启用 Batch。
-
如需获得创建作业所需的权限，请让您的管理员为您授予以下 IAM 角色：
-
项目的 Batch Job Editor (
roles/batch.jobsEditor) -
作业的服务账号（默认情况下为默认 Compute Engine 服务账号）的 Service Account User (
roles/iam.serviceAccountUser)
如需详细了解如何授予角色，请参阅管理对项目、文件夹和组织的访问权限。
-
项目的 Batch Job Editor (
创建使用 GPU 的作业
如需创建使用 GPU 的作业，请执行以下操作：
- 规划使用 GPU 的作业的要求。
- 根据您确定的要求和方法创建作业。 如需查看如何使用建议的选项创建作业的示例，请参阅本文档中的创建使用 GPU 的示例作业。
规划使用 GPU 的作业的要求
在创建使用 GPU 的作业之前，请按以下部分所述规划作业的要求：
第 1 步：选择 GPU 机器类型和消费选项
作业的要求因您偏好的 GPU 机器类型和使用选项而异，并且每种选项可能相互依赖。
如需为作业选择 GPU 机器类型和消耗选项，请完成以下步骤：
确定并选择您的优先事项：
- 如果您想优先考虑性能和基本价格，请参阅本文档中的选择 GPU 机器类型部分。
- 如果您想优先考虑资源可用性或折扣，请参阅本文档中的选择消费选项部分。
根据与您的第一个选择兼容的选项完成其余选择。
如需确定与您的第一个选择兼容的选项，请参阅 Compute Engine 文档中的按机器类型列出的消费选项可用性。
请参阅剩余部分，了解如何选择消费选项或选择 GPU 机器类型。
选择 GPU 机器类型
如需为作业选择 GPU 机器类型并了解其作业要求，请完成以下步骤：
选择 GPU 机器类型：如需查看可用的 GPU 机器类型（GPU 类型、GPU 数量和机器类型（vCPU 和内存）的有效组合）及其推荐的用例，请参阅 Compute Engine 文档中的 GPU 机器类型和加速器优化机器家族。
了解 GPU 机器类型的作业要求：作业指定 GPU 机器类型所需的字段因下表中的类别而异：
GPU 机器类型及其作业要求 加速器优化型虚拟机的 GPU：如果虚拟机的机器类型属于加速器优化型机器系列，则会自动挂接特定类型和数量的此类 GPU。
如需将 GPU 用于加速器优化虚拟机，我们建议您指定机器类型。每种加速器优化机器类型仅支持特定类型和数量的 GPU，因此无论您是否指定这些值，其功能都与加速器优化机器类型相同。
具体而言，Batch 还支持仅为经过加速器优化的虚拟机指定 GPU 的类型和数量，但由此产生的 vCPU 和内存选项通常非常有限。因此，我们建议您验证可用的 vCPU 和内存选项是否与作业的任务要求兼容。
N1 虚拟机的 GPU：这些 GPU 需要您指定要挂接到每个虚拟机的类型和数量，并且必须挂接到机器类型属于 N1 机器系列的虚拟机。
如需为 N1 虚拟机使用 GPU，建议您至少指定 GPU 的类型和数量。确保值组合与 N1 机器类型的有效 GPU 选项之一相匹配。对于使用任何特定类型和数量 GPU 的 N1 虚拟机，vCPU 和内存选项非常灵活。除非您使用 Google Cloud 控制台创建作业，否则您可以让 Batch 自动选择满足作业任务要求的机器类型。
选择使用选项
本部分总结了如何为作业选择消费选项，以及每种选项的作业要求。
下表介绍了 Batch 的可用消耗选项及其要求，并根据这些选项的用例（从大致最高到最低的资源可用性）列出了这些选项。
总而言之，我们建议大多数用户为作业选择以下消费选项：
- 除非满足其他条件之一，否则请使用灵活启动虚拟机（预览版）（如果满足您的要求，则首选此选项）或按需付费虚拟机。
- 如果您的作业需要极高的资源可用性保证，或者您有任何无法删除的未使用预留，请使用日历模式预留（预览版）（如果满足您的要求，则首选此模式）或预留。
- （可选）如果您的工作负载具有很强的容错能力，那么您或许可以尝试使用 Spot 虚拟机来降低费用。
| 使用选项及其作业要求 | |
|---|---|
预留
|
Batch 使用预留来运行可使用未使用的预留虚拟机的作业。如需详细了解使用预留的作业要求，请参阅使用虚拟机预留确保资源可用性页面。 |
日历模式下的预留（预览版）
|
Batch 会为满足以下所有条件的作业使用日历模式预留：
|
灵活启动虚拟机（预览版）
|
对于满足以下所有条件的作业，Batch 会使用灵活启动虚拟机：
|
按需
|
默认情况下，批处理作业对大多数其他作业使用按需模式。如果您想确保作业使用按需实例，请执行以下所有操作：1
1例外情况：对于使用 A3 系列机器的作业，默认行为是唯一的，我们建议您不要使用按需虚拟机，也不要将 |
Spot 虚拟机
|
Batch 会针对满足以下所有条件的作业使用 Spot 虚拟机：
|
第 2 步：安装 GPU 驱动程序
如需将 GPU 用于作业，您必须安装 GPU 驱动程序。如需安装 GPU 驱动程序，请选择以下方法之一：
自动安装 GPU 驱动程序（建议尽可能采用此方法）：如示例所示，若要让 Batch 从第三方位置提取所需的 GPU 驱动程序并代表您进行安装，请将作业的
installGpuDrivers字段设置为true。如果您的工作不需要您手动安装驱动程序，建议使用此方法。（可选）如果您需要指定 Batch 安装的 GPU 驱动程序版本，还可以设置
driverVersion字段。手动安装 GPU 驱动程序：如果满足以下任一条件，则必须使用此方法：
- 作业同时使用脚本和容器可运行对象，且没有互联网访问权限。如需详细了解作业拥有的访问权限，请参阅批量联网概览。
- 作业使用自定义虚拟机映像。如需详细了解虚拟机操作系统映像以及您可以使用的虚拟机操作系统映像，请参阅虚拟机操作系统环境概览。
如需手动安装所需的 GPU 驱动程序，建议采用以下方法：
创建包含 GPU 驱动程序的自定义虚拟机映像。
如需安装 GPU 驱动程序，请运行一个安装脚本，具体取决于您要使用的操作系统：
如果您的作业包含任何容器可运行对象，并且未使用 Container-Optimized OS，您还必须安装 NVIDIA 容器工具包
创建并提交使用 GPU 的作业时，指定包含 GPU 驱动程序的自定义虚拟机映像，并将作业的
installGpuDrivers字段设置为false（默认值）。
第 3 步：定义兼容的虚拟机资源
如需了解定义作业的虚拟机资源的要求和选项，请参阅作业资源。
总而言之，在为使用 GPU 的作业定义虚拟机资源时，您必须执行以下所有操作：
确保 GPU 机器类型在作业虚拟机的所在位置可用。
如需了解 GPU 机器类型的可用区域，请参阅 Compute Engine 文档中的加速器可用性。
如果您指定作业的机器类型，请确保该机器类型具有足够的 vCPU 和内存来满足作业的任务要求。无论何时使用 Google Cloud 控制台创建作业，都必须指定作业的机器类型；无论何时创建使用 GPU 的作业，都建议指定作业的机器类型，以便使用经过加速器优化的虚拟机。
请确保您使用有效方法为作业定义虚拟机资源：
- 使用
instances[].policy字段直接定义虚拟机资源（建议尽可能使用）。 此方法在示例中进行了演示。 - 使用
instances[].instanceTemplate字段通过模板定义虚拟机资源。 此方法是必需的，用于通过自定义映像手动安装 GPU 驱动程序。如需了解详情，请参阅使用虚拟机实例模板定义作业资源。
- 使用
创建使用 GPU 的作业示例
以下部分介绍了如何为每种消费选项创建示例作业。具体而言，示例作业使用建议的选项：全部自动安装 GPU 驱动程序，全部直接定义虚拟机资源。
| 使用选项 | 作业指令示例 |
|---|---|
| 预留或日历模式下的预留（预览版） | |
| 灵活启动虚拟机（预览版） | |
| 按需虚拟机或 Spot 虚拟机 |
根据您的 GPU 机器类型，选择以下选项之一： |
使用 GPU 和灵活启动虚拟机
您可以使用 gcloud CLI 或 Batch API 通过动态工作负载调度器创建使用 GPU 的 A3 虚拟机作业。
gcloud
创建一个 JSON 文件，用于安装 GPU 驱动程序、指定支持灵活启动虚拟机的 GPU 机器类型、阻止预留，并在支持该 GPU 机器类型的位置运行。
例如，如需创建使用 Flex-start 虚拟机和 GPU（适用于经过加速器优化的虚拟机）的基本脚本作业，请创建一个包含以下内容的 JSON 文件。
{ "taskGroups": [ { "taskSpec": { "runnables": [ { "script": { "text": "echo Hello world from task ${BATCH_TASK_INDEX}." } } ] }, "taskCount": 3, "parallelism": 1 } ], "allocationPolicy": { "instances": [ { "installGpuDrivers": INSTALL_GPU_DRIVERS, "policy": { "provisioningModel":"FLEX_START", "machineType": "MACHINE_TYPE", "reservation": "NO_RESERVATION" } } ], "location": { "allowedLocations": [ "ALLOWED_LOCATIONS" ] } }, "logsPolicy": { "destination": "CLOUD_LOGGING" } }替换以下内容：
INSTALL_GPU_DRIVERS：如果设置为true，Batch 会从第三方位置提取您在policy字段中指定的 GPU 类型所需的驱动程序，并代表您安装这些驱动程序。如果您将此字段设置为false（默认值），则需要手动安装 GPU 驱动程序才能使用任何 GPU 来运行此作业。MACHINE_TYPE：支持灵活启动虚拟机的 GPU 机器类型。如需了解详情，请参阅 Compute Engine 文档中的按机器类型列出的消费选项可用性。ALLOWED_LOCATIONS：您可以选择使用allowedLocations[]字段来指定作业的虚拟机允许运行的区域或区域中的特定可用区，例如，regions/us-central1允许在us-central1区域中的所有可用区运行。 请务必指定提供您希望用于此作业的 GPU 机器类型的位置。否则，如果您省略此字段，请确保作业的位置提供 GPU 机器类型。
如需创建并运行作业，请使用
gcloud alpha batch jobs submit命令：gcloud batch jobs submit JOB_NAME \ --location LOCATION \ --config JSON_CONFIGURATION_FILE替换以下内容：
JOB_NAME：作业的名称。LOCATION：作业的位置。JSON_CONFIGURATION_FILE：包含作业配置详细信息的 JSON 文件的路径。
API
向 v1alpha jobs.create 方法发出 POST 请求，该请求会安装 GPU 驱动程序、指定支持灵活启动虚拟机的 GPU 机器类型、阻止预留，并在支持 GPU 机器类型的位置运行。
例如，如需创建使用灵活启动虚拟机和 GPU（适用于加速器优化虚拟机）的基本脚本作业，请发出以下请求：
POST https://batch.googleapis.com/v1alpha/projects/PROJECT_ID/locations/LOCATION/jobs?job_id=JOB_NAME
{
"taskGroups": [
{
"taskSpec": {
"runnables": [
{
"script": {
"text": "echo Hello world from task ${BATCH_TASK_INDEX}."
}
}
]
},
"taskCount": 3,
"parallelism": 1
}
],
"allocationPolicy": {
"instances": [
{
"installGpuDrivers": INSTALL_GPU_DRIVERS,
"policy": {
"provisioningModel":"FLEX_START",
"machineType": "MACHINE_TYPE",
"reservation": "NO_RESERVATION"
}
}
],
"location": {
"allowedLocations": [
"ALLOWED_LOCATIONS"
]
}
},
"logsPolicy": {
"destination": "CLOUD_LOGGING"
}
}
替换以下内容：
PROJECT_ID：您的项目的项目 ID。LOCATION：作业的位置。JOB_NAME：作业的名称。INSTALL_GPU_DRIVERS：如果设置为true，Batch 会从第三方位置提取您在policy字段中指定的 GPU 类型所需的驱动程序，并代表您安装这些驱动程序。如果您将此字段设置为false（默认值），则需要手动安装 GPU 驱动程序才能将任何 GPU 用于此作业。MACHINE_TYPE：支持灵活启动虚拟机的 GPU 机器类型。如需了解详情，请参阅 Compute Engine 文档中的按机器类型列出的消费选项可用性。ALLOWED_LOCATIONS：您可以选择使用allowedLocations[]字段来指定作业的虚拟机允许运行的区域或区域中的特定可用区，例如，regions/us-central1允许在us-central1区域中的所有可用区运行。 请务必指定提供您希望用于此作业的 GPU 机器类型的位置。否则，如果您省略此字段，请确保作业的位置提供 GPU 机器类型。
为加速器优化型虚拟机使用 GPU
您可以使用Google Cloud 控制台、gcloud CLI、Batch API、Java、Node.js 或 Python 创建使用 GPU 的加速器优化虚拟机作业。
控制台
如需使用 Google Cloud 控制台创建使用 GPU 的作业，请执行以下操作：
在 Google Cloud 控制台中，前往作业列表页面。
点击 创建。 系统会打开创建批处理作业页面。在左侧窗格中，作业详情页面处于选中状态。
配置作业详情页面：
可选：在作业名称字段中，自定义作业名称。
例如，输入
example-gpu-job。配置任务详情部分：
在新的可运行对象窗口中，为此作业添加至少一个要运行的脚本或容器。
例如，如需创建基本脚本作业，请执行以下操作：
选中脚本复选框。系统会显示一个字段。
在该字段中，输入以下脚本：
echo Hello world from task ${BATCH_TASK_INDEX}.点击完成。
在任务数字段中，输入相应作业的任务数。
例如，输入
3。可选：在并行性字段中，输入要并发运行的任务数。
例如，输入
1（默认值）。
配置资源规范页面：
在左侧窗格中，点击资源规范。 系统会打开资源规范页面。
在虚拟机预配模型部分中，为相应作业的虚拟机选择以下使用选项之一：
如果您的作业可以承受抢占，并且您想要使用折扣价的 Spot 虚拟机，请选择 Spot。
否则，如需使用按需付费型虚拟机，请选择标准（默认）。
为此作业选择位置。
在区域字段中，选择一个区域。
在可用区字段中，执行以下操作之一：
如果您希望将此作业限制为仅在特定可用区中运行，请选择一个可用区。
否则，请选择任意（默认）。
为相应作业的虚拟机选择 GPU 机器类型：
在机器系列选项中，点击 GPU。
在 GPU 类型字段中，选择 GPU 类型。然后，在 GPU 数量字段中，选择每个虚拟机的 GPU 数量。
如果您选择的是加速器优化型虚拟机的 GPU 类型，则机器类型字段仅允许根据您选择的 GPU 类型和数量选择一种机器类型。
如需自动安装 GPU 驱动程序，请选择 GPU 驱动程序安装（默认）。
配置每项任务所需的虚拟机资源量：
在核心数字段中，输入每个任务的 vCPUs 数量。
例如，输入
1（默认值）。在内存字段中，输入每个任务的 RAM 量（以 GB 为单位）。
例如，输入
0.5（默认值）。
点击完成。
可选：配置此作业的其他字段。
可选：如需查看作业配置，请在左侧窗格中点击预览。
点击创建。
作业详情页面会显示您创建的作业。
gcloud
创建一个 JSON 文件，用于安装 GPU 驱动程序、指定加速器优化机器系列中的机器类型，并在具有 GPU 机器类型的位置运行。
例如，如需创建使用 GPU 的基本脚本作业（适用于经过加速器优化的虚拟机），请创建一个包含以下内容的 JSON 文件：
{ "taskGroups": [ { "taskSpec": { "runnables": [ { "script": { "text": "echo Hello world from task ${BATCH_TASK_INDEX}." } } ] }, "taskCount": 3, "parallelism": 1 } ], "allocationPolicy": { "instances": [ { "installGpuDrivers": INSTALL_GPU_DRIVERS, "policy": { "provisioningModel": "PROVISIONING_MODEL", "machineType": "MACHINE_TYPE", "reservation": "NO_RESERVATION" } } ], "location": { "allowedLocations": [ "ALLOWED_LOCATIONS" ] } }, "logsPolicy": { "destination": "CLOUD_LOGGING" } }替换以下内容：
INSTALL_GPU_DRIVERS：如果设置为true，Batch 会从第三方位置提取您在policy字段中指定的 GPU 类型所需的驱动程序，并代表您将其安装。如果您将此字段设置为false（默认值），则需要手动安装 GPU 驱动程序才能将任何 GPU 用于此作业。PROVISIONING_MODEL：消费选项的预配模型：STANDARD表示按需，SPOT表示 Spot 虚拟机。MACHINE_TYPE：加速器优化机器系列中的机器类型。ALLOWED_LOCATIONS：您可以选择使用allowedLocations[]字段来指定作业的虚拟机允许运行的区域或区域中的特定可用区，例如，regions/us-central1允许在us-central1区域中的所有可用区运行。 请务必指定提供您希望用于此作业的 GPU 机器类型的位置。否则，如果您省略此字段，请确保作业的位置提供 GPU 机器类型。
如需创建并运行作业，请使用
gcloud batch jobs submit命令：gcloud batch jobs submit JOB_NAME \ --location LOCATION \ --config JSON_CONFIGURATION_FILE替换以下内容：
JOB_NAME：作业的名称。LOCATION：作业的位置。JSON_CONFIGURATION_FILE：包含作业配置详细信息的 JSON 文件的路径。
API
向 jobs.create 方法发出 POST 请求，该请求会安装 GPU 驱动程序、指定加速器优化机器系列中的机器类型，并在具有 GPU 机器类型的位置运行。
例如，如需创建使用 GPU 的基本脚本作业（适用于经过加速器优化的虚拟机），请发出以下请求：
POST https://batch.googleapis.com/v1/projects/PROJECT_ID/locations/LOCATION/jobs?job_id=JOB_NAME
{
"taskGroups": [
{
"taskSpec": {
"runnables": [
{
"script": {
"text": "echo Hello world from task ${BATCH_TASK_INDEX}."
}
}
]
},
"taskCount": 3,
"parallelism": 1
}
],
"allocationPolicy": {
"instances": [
{
"installGpuDrivers": INSTALL_GPU_DRIVERS,
"policy": {
"provisioningModel": "PROVISIONING_MODEL",
"machineType": "MACHINE_TYPE",
"reservation": "NO_RESERVATION"
}
}
],
"location": {
"allowedLocations": [
"ALLOWED_LOCATIONS"
]
}
},
"logsPolicy": {
"destination": "CLOUD_LOGGING"
}
}
替换以下内容：
PROJECT_ID：您的项目的项目 ID。LOCATION：作业的位置。JOB_NAME：作业的名称。INSTALL_GPU_DRIVERS：如果设置为true，Batch 会从第三方位置提取您在policy字段中指定的 GPU 类型所需的驱动程序，并代表您安装这些驱动程序。如果您将此字段设置为false（默认值），则需要手动安装 GPU 驱动程序才能将任何 GPU 用于此作业。PROVISIONING_MODEL：消费选项的预配模型：STANDARD表示按需，SPOT表示 Spot 虚拟机。MACHINE_TYPE：加速器优化机器系列中的机器类型。ALLOWED_LOCATIONS：您可以选择使用allowedLocations[]字段来指定作业的虚拟机允许运行的区域或区域中的特定可用区，例如，regions/us-central1允许在us-central1区域中的所有可用区运行。 请务必指定提供您希望用于此作业的 GPU 机器类型的位置。否则，如果您省略此字段，请确保作业的位置提供 GPU 机器类型。
Java
Node.js
Python
为 N1 虚拟机使用 GPU
您可以使用 Google Cloud 控制台、gcloud CLI、Batch API、Java、Node.js 或 Python 创建使用 GPU 的 N1 虚拟机作业。
控制台
如需使用 Google Cloud 控制台创建使用 GPU 的作业，请执行以下操作：
在 Google Cloud 控制台中，前往作业列表页面。
点击 创建。 系统会打开创建批处理作业页面。在左侧窗格中，作业详情页面处于选中状态。
配置作业详情页面：
可选：在作业名称字段中，自定义作业名称。
例如，输入
example-gpu-job。配置任务详情部分：
在新的可运行对象窗口中，为此作业添加至少一个要运行的脚本或容器。
例如，如需创建基本脚本作业，请执行以下操作：
选中脚本复选框。系统会显示一个字段。
在该字段中，输入以下脚本：
echo Hello world from task ${BATCH_TASK_INDEX}.点击完成。
在任务数字段中，输入相应作业的任务数。
例如，输入
3。可选：在并行性字段中，输入要并发运行的任务数。
例如，输入
1（默认值）。
配置资源规范页面：
在左侧窗格中，点击资源规范。 系统会打开资源规范页面。
在虚拟机预配模型部分中，为相应作业的虚拟机选择以下使用选项之一：
如果您的作业可以承受抢占，并且您想要使用折扣价的 Spot 虚拟机，请选择 Spot。
否则，如需使用按需付费型虚拟机，请选择标准（默认）。
为此作业选择位置。
在区域字段中，选择一个区域。
在可用区字段中，执行以下操作之一：
如果您希望将此作业限制为仅在特定可用区中运行，请选择一个可用区。
否则，请选择任意（默认）。
为相应作业的虚拟机选择 GPU 机器类型：
在机器系列选项中，点击 GPU。
在 GPU 类型字段中，选择 GPU 类型。
如果您选择了 N1 虚拟机的 GPU 类型之一，则系列字段会设置为 N1。
在 GPU 数量字段中，为每个虚拟机选择 GPU 数量。
在机器类型字段中，选择机器类型。
如需自动安装 GPU 驱动程序，请选择 GPU 驱动程序安装（默认）。
配置每项任务所需的虚拟机资源量：
在核心数字段中，输入每个任务的 vCPUs 数量。
例如，输入
1（默认值）。在内存字段中，输入每个任务的 RAM 量（以 GB 为单位）。
例如，输入
0.5（默认值）。
点击完成。
可选：配置此作业的其他字段。
可选：如需查看作业配置，请在左侧窗格中点击预览。
点击创建。
作业详情页面会显示您创建的作业。
gcloud
创建一个 JSON 文件，该文件用于安装 GPU 驱动程序、定义
accelerators[]字段的type和count子字段，并在具有 GPU 机器类型的位置运行。例如，如需创建使用 GPU 的 N1 虚拟机的基本脚本作业，并让 Batch 选择确切的 N1 机器类型，请创建一个包含以下内容的 JSON 文件：
{ "taskGroups": [ { "taskSpec": { "runnables": [ { "script": { "text": "echo Hello world from task ${BATCH_TASK_INDEX}." } } ] }, "taskCount": 3, "parallelism": 1 } ], "allocationPolicy": { "instances": [ { "installGpuDrivers": INSTALL_GPU_DRIVERS, "policy": { "provisioningModel": "PROVISIONING_MODEL", "reservation": "NO_RESERVATION", "accelerators": [ { "type": "GPU_TYPE", "count": GPU_COUNT } ] } } ], "location": { "allowedLocations": [ "ALLOWED_LOCATIONS" ] } }, "logsPolicy": { "destination": "CLOUD_LOGGING" } }替换以下内容：
INSTALL_GPU_DRIVERS：如果设置为true，Batch 会从第三方位置提取您在policy字段中指定的 GPU 类型所需的驱动程序，并代表您将其安装。如果您将此字段设置为false（默认值），则需要手动安装 GPU 驱动程序才能使用任何 GPU 来运行此作业。PROVISIONING_MODEL：消费选项的预配模型：STANDARD表示按需，SPOT表示 Spot 虚拟机。GPU_TYPE：GPU 类型。您可以使用gcloud compute accelerator-types list命令查看可用 GPU 类型的列表。仅将此字段用于 N1 虚拟机的 GPU。GPU_COUNT：指定类型的 GPU 数量。如需详细了解有效选项，请参阅 N1 机器系列的 GPU 机器类型。仅将此字段用于 N1 虚拟机的 GPU。ALLOWED_LOCATIONS：您可以选择使用allowedLocations[]字段来指定作业的虚拟机允许运行的区域或区域中的特定可用区，例如，regions/us-central1允许在us-central1区域中的所有可用区运行。 请务必指定提供您希望用于此作业的 GPU 机器类型的位置。否则，如果您省略此字段，请确保作业的位置提供 GPU 机器类型。
如需创建并运行作业，请使用
gcloud batch jobs submit命令：gcloud batch jobs submit JOB_NAME \ --location LOCATION \ --config JSON_CONFIGURATION_FILE替换以下内容：
JOB_NAME：作业的名称。LOCATION：作业的位置。JSON_CONFIGURATION_FILE：包含作业配置详细信息的 JSON 文件的路径。
API
向 jobs.create 方法发出 POST 请求，该请求会安装 GPU 驱动程序，定义 accelerators[] 字段的 type 和 count 子字段，并使用具有 GPU 机器类型的位置。
例如，如需创建使用 GPU 的 N1 虚拟机并让 Batch 选择确切的 N1 机器类型，请发出以下请求：
POST https://batch.googleapis.com/v1/projects/PROJECT_ID/locations/LOCATION/jobs?job_id=JOB_NAME
{
"taskGroups": [
{
"taskSpec": {
"runnables": [
{
"script": {
"text": "echo Hello world from task ${BATCH_TASK_INDEX}."
}
}
]
},
"taskCount": 3,
"parallelism": 1
}
],
"allocationPolicy": {
"instances": [
{
"installGpuDrivers": INSTALL_GPU_DRIVERS,
"policy": {
"provisioningModel": "PROVISIONING_MODEL",
"reservation": "NO_RESERVATION",
"accelerators": [
{
"type": "GPU_TYPE",
"count": GPU_COUNT
}
]
}
}
],
"location": {
"allowedLocations": [
"ALLOWED_LOCATIONS"
]
}
},
"logsPolicy": {
"destination": "CLOUD_LOGGING"
}
}
替换以下内容：
PROJECT_ID：您的项目的项目 ID。LOCATION：作业的位置。JOB_NAME：作业的名称。INSTALL_GPU_DRIVERS：如果设置为true，Batch 会从第三方位置提取您在policy字段中指定的 GPU 类型所需的驱动程序，并代表您安装这些驱动程序。如果您将此字段设置为false（默认值），则需要手动安装 GPU 驱动程序才能使用任何 GPU 来运行此作业。PROVISIONING_MODEL：消费选项的预配模型：STANDARD表示按需，SPOT表示 Spot 虚拟机。GPU_TYPE：GPU 类型。您可以使用gcloud compute accelerator-types list命令查看可用 GPU 类型的列表。仅将此字段用于 N1 虚拟机的 GPU。GPU_COUNT：指定类型的 GPU 数量。如需详细了解有效选项，请参阅 N1 机器系列的 GPU 机器类型。 仅将此字段用于 N1 虚拟机的 GPU。ALLOWED_LOCATIONS：您可以选择使用allowedLocations[]字段来指定作业的虚拟机允许运行的区域或区域中的特定可用区，例如，regions/us-central1允许在us-central1区域中的所有可用区运行。 请务必指定提供您希望用于此作业的 GPU 机器类型的位置。否则，如果您省略此字段，请确保作业的位置提供 GPU 机器类型。