在没有GCS的情况下使用colab TPU

import tensorflow as tf
try:
    tpu = tf.distribute.cluster_resolver.TPUClusterResolver()  # TPU detection
    tf.config.experimental_connect_to_cluster(tpu)
    tf.tpu.experimental.initialize_tpu_system(tpu)
    tpu_strategy = tf.distribute.experimental.TPUStrategy(tpu)
    print('TPU device found')
except ValueError:
    tpu_strategy = None
    print('No TPU device found')

with tpu_strategy.scope():
    model = tf.keras.Sequential([...])  # 定义模型结构
    model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])  # 编译模型
    model.fit(x_train, y_train, epochs=10, validation_data=(x_val, y_val))  # 使用TPU进行模型训练

predictions = model.predict(x_test)