【发布时间】:2016-05-18 09:52:00
【问题描述】:
正如标题所说,我想找出 tensorflow 能够检测或使用的设备(cpus 或 gpus)的数量。
这个问题的原因是因为我在具有多个 cpu 的机器上运行带有 tf 后端的 keras,但它在运行时只使用一个 CPU。
【问题讨论】:
标签: python c++ machine-learning tensorflow
正如标题所说,我想找出 tensorflow 能够检测或使用的设备(cpus 或 gpus)的数量。
这个问题的原因是因为我在具有多个 cpu 的机器上运行带有 tf 后端的 keras,但它在运行时只使用一个 CPU。
【问题讨论】:
标签: python c++ machine-learning tensorflow
查看这个关于 CPU 扩展的错误报告; https://github.com/tensorflow/tensorflow/issues/583
据说它已通过 2016 年 5 月初发布的更改 ab02c5 解决。尝试从源代码构建 tensorflow 以获取所有最新更新,这些更新很可能不在稳定版本中。
【讨论】: