I have created a tpu compute instance on Google cloud that I can access by remote ssh. When I ssh in and run my program it prints out the list of available devices like this
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
Unfortunately it shows a CPU device but no TPU device, so TF appears to be running on the CPU. Can anyone tell me how to get TF to see the TPU device?