googleapis / googleapis/python-aiplatform

Vertex AI Pipelines using TPUs Internal Error encountered

Abierto
#2,231 0 comentarios 0 reacciones 0 asignados Ver en GitHub
api: vertex-ai
Lenguaje dominante
Python
Estrellas
905
Forks
465
Merge medio
1 d 13 h
PR fusionados (30 d)
44

Descripción

#### 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.

Guía de contribución

Abrir la guía de contribución

Evaluación

Este issue todavía no se ha evaluado.

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.