在google colab中，有没有一种方法可以检查TPU版本正在运行？

import tensorflow as tf

# 检查是否存在 TPU
if 'COLAB_TPU_ADDR' not in os.environ:
    print('TPU 没有连接')
else:
    tpu_address = 'grpc://' + os.environ['COLAB_TPU_ADDR']
    tpu_resolver = tf.distribute.cluster_resolver.TPUClusterResolver(tpu=tpu_address)
    tf.config.experimental_connect_to_cluster(tpu_resolver)
    tf.tpu.experimental.initialize_tpu_system(tpu_resolver)
    tpu_version = tf.tpu.experimental.initialize_tpu_system(tpu_resolver).split(' ')[1]
    print('TPU 版本:', tpu_version)