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

m0_49630462

已于 2023-03-04 20:30:42 修改

阅读量3.2k

点赞数 1

文章标签： tensorflow 深度学习 python

于 2022-11-16 16:18:38 首次发布

本文链接：https://blog.csdn.net/m0_49630462/article/details/127887484

版权

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__)