Public

gistfile1.txt

Hello everyone I am trying to run the following pipeline with TPUs,

In GCP I have quotas, and credits, yet in vertex it fails with the same message: "Internal Error" please contact Vertex AI, but to contact them, yet does anyone see anything wrong with the following definition?


 
 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="CENSORED",
         location="us-central1",
     )
 
 
 if __name__ == "__main__":
     import kfp.compiler as compiler
 
     compiler.Compiler().compile(pipeline, "problem_pipeline.yaml")