如何防止TPUEstimator使用GPU或TPU

import tensorflow as tf

# 禁用GPU和TPU
tf.config.set_visible_devices([], 'GPU')
tf.config.set_visible_devices([], 'TPU')

import tensorflow as tf

# 忽略TPU设备
tpu_estimator = tf.estimator.TPUEstimator(
    model_fn=model_fn,
    config=tpu_config,
    use_tpu=False
)

import tensorflow as tf

# 忽略GPU设备
session_config = tf.ConfigProto(device_count={'GPU': 0})
tpu_estimator = tf.estimator.TPUEstimator(
    model_fn=model_fn,
    config=tpu_config,
    train_batch_size=batch_size,
    eval_batch_size=batch_size,
    predict_batch_size=batch_size,
    session_config=session_config
)