正如标题所说,我想找出 tensorflow 能够检测或使用的设备(cpus 或 gpus)的数量。
这个问题的原因是因为我在一台有多个 cpu 的机器上运行带有 tf 后端的 keras,但它在运行时只使用一个 CPU。
最佳答案
查看有关 CPU 缩放的错误报告; https://github.com/tensorflow/tensorflow/issues/583
据推测,它已通过 2016 年 5 月初发布的更改 ab02c5 得到解决。尝试从源代码构建 tensorflow 以获取所有最新更新,这些更新很可能不在稳定版本中。
关于python - 在 tensorflow 中查找检测到的设备数量,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37296064/