tensorflow 数据训练时，查看设备使用情况。

训练数据时候发现跑的很慢，查看一下是使用什么跑的

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())　

输出：

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 6321505847024071618
]

原来是cpu在跑，怪不得这么慢！