tensorflow-gpu测试代码以及版本检测

tensorflow-gpu测试代码以及版本检测

废话不说：直接上代码

import tensorflow as tf
tf.config.list_physical_devices('GPU')
#查看tensorflow版本  #gpu环境测试
print(tf.__version__)

a = tf.constant(2.0)
b = tf.constant(4.0)
print(a + b)

print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

#输出显示测试安装结果
print('GPU',tf.test.is_gpu_available())

# 此命令为获取安装的tensorflow版本
print("tf.__version__ = ",tf.__version__) # 输出版本

#查看tensorflow安装路径
print("tf.__path__  = ",tf.__path__)