googleapis / googleapis/python-aiplatform
Vertex AI Pipelines using TPUs Internal Error encountered
- 主要语言
- Python
- 星标
- 905
- 派生
- 465
- 平均合并
- 1 天 13 小时
- 30 天内合并 PR
- 44
描述
#### Environment details
- Python version: 3.11
- pip version: 23.1
- `google-cloud-aiplatform` version: 1.25.0
- kfp=2.0.0rc1 (all versions tested up to 2.0.0b16)
#### Steps to reproduce
1. This pipeline:
```
from kfp import dsl
import google_cloud_pipeline_components.v1.custom_job as gcp_components_custom_job
@dsl.component
def print_test():
print("Hello World!")
# Using TPUs
printOp = gcp_components_custom_job.create_custom_training_job_from_component(
print_test,
display_name="Print Test",
machine_type="cloud-tpu",
accelerator_type="TPU_V3",
accelerator_count=8,
)
@dsl.pipeline
def pipeline():
print_task = printOp(
project="REDACTED",
location="us-central1",
)
if __name__ == "__main__":
import kfp.compiler as compiler
compiler.Compiler().compile(pipeline, "problem_pipeline.yaml")
```
2. send pipeline to run in vertex
I end up with 40 min laters with a lot of internal errors and contact Vertex:

Thanks!
I have the quota for this machines and I have no idea what more to provide.
贡献指南
调研方向
从提供的 Python pipeline 和对 compiler.Compiler().compile 的调用开始,然后使用所述的 TPU 设置在 Vertex AI 中运行编译后的 problem_pipeline.yaml。将运行输出和内部错误与环境版本及复现步骤进行比较;当 TPU pipeline 在没有报告的内部错误的情况下运行,或将失败范围缩小到一个有文档记录的外部服务问题时,即表示完成。
由索引模型根据 Issue 内容生成。
评估
- 技术栈
- google-cloud, python
- 领域
- cloud, machine-learning
- Issue 类型
- 缺陷
- 难度
- 4/5
- 预计耗时
- 3-5 天
- 活跃度
- 停滞
- 描述清晰度
- 需要澄清
- 新手友好度
- 35/100