查找tensorflow中检测到的设备数量

Question

正如标题所说，我想找出张量流能够检测或使用的设备数量（cpus或gpus）。

这个问题的原因是因为我在具有多个cpu的机器上运行带有tf后端的keras，但它在运行时只使用一个CPU。

Answer 1

查看有关CPU扩展的错误报告; https://github.com/tensorflow/tensorflow/issues/583

据推测，2016年5月初发布的变更ab02c5解决了这个问题。尝试从源代码构建张量流以获取所有最新更新，这些更新很可能不在稳定版本中。