测试TensorFlow2.2GPU是否可用

#测试TensorFlow-GPU是否可用
import tensorflow as tf
import os
print(tf.__version__)
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'  # 不显示等级2以下的提示信息
print('GPU', tf.config.list_physical_devices('GPU'))#列出所有的GPU驱动号
a = tf.constant(2.0)
b = tf.constant(4.0)
print(a + b)

输出如下：