使用虚拟机实例模板定义作业资源

本文档介绍如何通过指定 Compute Engine 虚拟机实例模板为批量作业定义资源。

（可选）您可以使用以下方法之一为作业定义资源：

使用 policy 字段直接定义作业的资源，如创建基本作业中所述。
按照本部分的说明，使用 instanceTemplate 字段通过指定虚拟机实例模板来定义作业的资源。创建使用非默认虚拟机映像的作业时，必须使用此方法。

准备工作

如果您之前未使用过 Batch，请查看 Batch 使用入门，并完成项目和用户前提条件以启用 Batch。
如需获取创建作业所需的权限，请让管理员向您授予以下 IAM 角色：
- 项目的 Batch Job Editor (roles/batch.jobsEditor)
- 作业的服务帐号的 Service Account User (roles/iam.serviceAccountUser)，默认为默认 Compute Engine 服务帐号
- 通过 Compute Engine 虚拟机实例模板创建作业：对虚拟机实例模板执行 Compute Viewer (roles/compute.viewer) 操作
如需详细了解如何授予角色，请参阅管理访问权限。

您也可以通过自定义角色或其他预定义角色来获取所需的权限。

使用 Compute Engine 虚拟机实例模板创建作业

本部分提供了有关如何基于现有虚拟机实例模板创建基本脚本作业的示例。您可以使用 gcloud CLI、Batch API、Go、Java、Node.js、Python 或 C++ 通过虚拟机实例模板创建作业。

gcloud

如需使用 gcloud CLI 基于虚拟机实例模板创建作业，请使用 gcloud batch jobs submit 命令，并在作业的 JSON 配置文件中指定虚拟机实例模板。

例如，如需根据虚拟机实例模板创建基本脚本作业，请运行以下命令：

在当前目录中创建一个名为 hello-world-instance-template.json 且包含以下内容的 JSON 文件：

{
    "taskGroups": [
        {
            "taskSpec": {
                "runnables": [
                    {
                        "script": {
                            "text": "echo Hello world! This is task ${BATCH_TASK_INDEX}. This job has a total of ${BATCH_TASK_COUNT} tasks."
                        }
                    }
                ],
                "computeResource": {
                    "cpuMilli": 2000,
                    "memoryMib": 16
                },
                "maxRetryCount": 2,
                "maxRunDuration": "3600s"
            },
            "taskCount": 4,
            "parallelism": 2
        }
    ],
    "allocationPolicy": {
        "instances": [
            {
                "installGpuDrivers" INSTALL_GPU_DRIVERS,
                "instanceTemplate": "INSTANCE_TEMPLATE_NAME"
            }
        ]
    },
    "labels": {
        "department": "finance",
        "env": "testing"
    },
    "logsPolicy": {
        "destination": "CLOUD_LOGGING"
    }
}

替换以下内容：

INSTALL_GPU_DRIVERS：可选。如果设置为 true，Batch 会提取您在 Compute Engine 虚拟机实例模板中指定的 GPU 类型所需的驱动程序，然后 Batch 会代表您安装这些驱动程序。如需了解详情，请参阅如何创建使用 GPU 的作业。
INSTANCE_TEMPLATE_NAME：现有 Compute Engine 虚拟机实例模板的名称。了解如何创建和列出实例模板。

运行以下命令：

gcloud batch jobs submit example-template-job \
  --location us-central1 \
  --config hello-world-instance-template.json

API

如需使用 Batch API 创建基本作业，请使用 jobs.create 方法，并在 allocationPolicy 字段中指定虚拟机实例模板。

例如，如需基于虚拟机实例模板创建基本脚本作业，请使用以下请求：

POST https://batch.googleapis.com/v1/projects/PROJECT_ID/locations/us-central1/jobs?job_id=example-script-job

{
    "taskGroups": [
        {
            "taskSpec": {
                "runnables": [
                    {
                        "script": {
                            "text": "echo Hello world! This is task ${BATCH_TASK_INDEX}. This job has a total of ${BATCH_TASK_COUNT} tasks."
                        }
                    }
                ],
                "computeResource": {
                    "cpuMilli": 2000,
                    "memoryMib": 16
                },
                "maxRetryCount": 2,
                "maxRunDuration": "3600s"
            },
            "taskCount": 4,
            "parallelism": 2
        }
    ],
    "allocationPolicy": {
        "instances": [
            {
                "installGpuDrivers" INSTALL_GPU_DRIVERS,
                "instanceTemplate": "INSTANCE_TEMPLATE_NAME"
            }
        ]
    },
    "labels": {
        "department": "finance",
        "env": "testing"
    },
    "logsPolicy": {
        "destination": "CLOUD_LOGGING"
    }
}

替换以下内容：

PROJECT_ID：您的项目 ID。
INSTALL_GPU_DRIVERS：可选。如果设置为 true，Batch 会提取您在 Compute Engine 虚拟机实例模板中指定的 GPU 类型所需的驱动程序，然后 Batch 会代表您安装这些驱动程序。如需了解详情，请参阅如何创建使用 GPU 的作业。
INSTANCE_TEMPLATE_NAME：现有 Compute Engine 虚拟机实例模板的名称。了解如何创建和列出实例模板。