import tensorflow as tf def print_gpu_availability(): """打印Tensorflow的GPU可用性""" gpus = tf.config.list_physical_devices('GPU') if gpus: print("GPU is available") else: print("GPU is not available")