tensorflow学习之路(5):利用tensorflow输出devices信息

利用tensorflow写相关项目时，尤其是写多GPU项目时，需要指定某个GPU，利用下面代码：

with tf.device('/gpu:0'):
    ....
with tf.device('/gpu:1'):
    ...
with tf.device('/cpu:0'):
    ...

但是，往往不知道GUP或者CUP相关信息，那么，可以通过下面的代码查看对应的信息：

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

例如，输出结果就是：

[name: "/cpu:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 3859781045412664391
]