Vertex AI自定义作业ID：手动指定及容器内访问方法问询

1. 启动自定义作业时能否手动指定Custom job ID？ #

可以手动指定。Vertex AI支持在创建自定义作业时直接设置自定义的作业ID，只需遵循以下命名规则：

• 仅包含小写字母、数字和连字符（-）
• 以字母开头
• 长度不超过63个字符
• 不能以连字符结尾

具体实现方式： #

使用gcloud命令行工具

添加--job-id参数指定自定义ID：

gcloud ai custom-jobs create --job-id=my-manual-job-id --region=us-central1 --image-uri=gcr.io/your-project/your-custom-image

使用Python SDK

在CustomJob构造函数中设置job_id参数：

from google.cloud import aiplatform

aiplatform.init(project="your-project-id", region="us-central1")

custom_job = aiplatform.CustomJob(
    display_name="your-job-name",
    job_id="my-manual-job-id",
    container_uri="gcr.io/your-project/your-custom-image",
)

custom_job.submit()

2. 如何从自定义容器内部访问自动生成的作业ID？ #

如果未手动指定作业ID，Vertex AI会自动生成唯一ID，并将其注入到容器的环境变量中，变量名为AIP_JOB_ID。你可以直接在容器内的代码中读取该变量：

Python示例： #

import os

auto_generated_job_id = os.environ.get("AIP_JOB_ID")
print(f"自动生成的作业ID: {auto_generated_job_id}")

Shell脚本示例： #

echo "自动生成的作业ID: $AIP_JOB_ID"

内容的提问来源于stack exchange，提问作者Stiefel