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_libprint(device_lib.list_local_devices())

例如，输出结果就是：

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