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

find number of detected devices in tensorflow

本文关键字：tensorflow 查找检测更新时间：2023-10-16

正如标题所说，我想了解tensorflow能够检测或使用的设备（cpu或gpu）的数量。

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

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

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