0

我正在构建一个包含以下训练步骤的管道：

(train(
    model=pretrained_model,
    unique_tags=preprocess_output.outputs["unique_tags_path"],
    train_encodings=preprocess_output.outputs["train_encodings_path"],
    train_labels=preprocess_output.outputs["train_labels_path"],
).
add_node_selector_constraint('cloud.google.com/gke-accelerator', 'nvidia-tesla-k80').
set_gpu_limit(1))

如文档中所述，为管道步骤指定机器类型

使用 Vertex AI 管道执行时，我无法访问请求的 GPU 并收到以下错误：

com.google.cloud.ai.platform.common.errors.AiPlatformException: code=INVALID_ARGUMENT,
message=Accelerator "NVIDIA_TESLA_K80" is not supported for machine type "n1-highmem-2",
cause=null; Failed to create custome job for the task.

如何获取有关可用于 Vertex AI 管道的 GPU 机器的信息？我想避免使用发送培训作业的组件。

谢谢

4

0 回答 0