查询自己的TensorFlow是否是gpu版本

import tensorflow as tf

# 输出 TensorFlow 版本信息
print("TensorFlow version:", tf.__version__)

# 输出 TensorFlow 是否支持 GPU
print("GPU available:", tf.config.list_physical_devices('GPU'))

# 输出 TensorFlow 是否使用 GPU 版本
print("Using GPU version:", tf.test.is_built_with_cuda())