googleapis / googleapis/python-aiplatform

Vertex AI Pipelines using TPUs Internal Error encountered

オープン
#2,231 コメント 0 件 リアクション 0 件 担当者 0 名 GitHub で見る
api: vertex-ai
主要言語
Python
スター
905
フォーク
465
平均マージ
1日 13時間
マージ済み PR(30日)
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:

![image](https://github.com/googleapis/python-aiplatform/assets/3343006/e7be22e6-1b3f-4a65-8695-39190ad394e7)

Thanks!

I have the quota for this machines and I have no idea what more to provide.

コントリビューションガイド

コントリビューションガイドを開く

評価

この issue はまだ評価されていません。

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。