tensorflow-gpu 2.0后查看是否启用GPU的指令

tensorflow-gpu 2.0后查看是否启用GPU的指令

前言
因为之前用查来的print(tensorflow.test.is_gpu_available())方法执行后发现反馈为 is_gpu_available (from tensorflow.python.framework.test_util) is deprecated and will be removed in a future version.即该方法已经过时并且将在之后的版本删除。本着对自己爱折腾以后多半还会重新装环境的认知，记录下用其提示的新方法以备用。

正文

环境里安装好tensorflow后，进入python，然后输入

import tensorflow as tf
tf.config.list_physical_devices('GPU')

如果GPU正常运行成功则会显示：

[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

不同设备可能有所不同，不过总之要有device_type='GPU'这部分应该就OK了。