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

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

废话不说：直接上代码

import tensorflow as tftf.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__)

如果觉得《tensorflow-gpu测试代码以及版本检测》对你有帮助，请点赞、收藏，并留下你的观点哦！