No TPU platform registered.

[asikuna]

I'm trying to run my Keras model on my TPU Pod and it's showing this error:

2023-03-30 00:16:14.818502: I tensorflow/compiler/xla/stream_executor/tpu/tpu_platform_interface.cc:76] No TPU platform registered. Waiting 1 second and trying again... (4 tries left)
2023-03-30 00:16:15.818648: I tensorflow/compiler/xla/stream_executor/tpu/tpu_platform_interface.cc:76] No TPU platform registered. Waiting 1 second and trying again... (3 tries left)
2023-03-30 00:16:16.818815: I tensorflow/compiler/xla/stream_executor/tpu/tpu_platform_interface.cc:76] No TPU platform registered. Waiting 1 second and trying again... (2 tries left)
2023-03-30 00:16:17.818983: I tensorflow/compiler/xla/stream_executor/tpu/tpu_platform_interface.cc:76] No TPU platform registered. Waiting 1 second and trying again... (1 tries left)
2023-03-30 00:16:18.819142: I tensorflow/compiler/xla/stream_executor/tpu/tpu_platform_interface.cc:73] No TPU platform found.

 Not really sure what to do. I already tried setting the environment variables for TPU_NAME and TPU_LOAD_LIBRARY. I also set the name manually and it's still showing the error.

[Aris_O]

Hi @asikuna 

Welcome back to Google Cloud Community.

Cloud TPU supports two TPU architectures: TPU VM or the TPU Node. The two VM architectures are described in System Architecture. The troubleshooting procedures in this document apply to both configurations, but there is also a separate TPU VM section specifically for troubleshooting problems that only occur with TPU VM configurations.

This reference might help you:
https://cloud.google.com/tpu/docs/troubleshooting/trouble-tf#trouble-connecting
https://cloud.google.com/tpu/docs/troubleshooting/trouble-tf