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

Question

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

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

Answer 1

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

它应该在2016年5月初发布的更改ab02c5中得到解决。尝试从源代码构建tensorflow以获取所有最新更新，这些更新很可能不在稳定版本中。