在Google Colab中，有没有一种编程方法可以检查我连接的是gpu或tpu环境？

import tensorflow as tf

if tf.test.gpu_device_name():
    print('GPU环境已连接')
else:
    print('未连接到GPU环境')

try:
    tpu = tf.distribute.cluster_resolver.TPUClusterResolver()
    tf.config.experimental_connect_to_cluster(tpu)
    tf.tpu.experimental.initialize_tpu_system(tpu)
    print('TPU环境已连接')
except:
    print('未连接到TPU环境')