gpt4 book ai didi

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

转载 作者:太空狗 更新时间:2023-10-29 22:59:17 25 4
gpt4 key购买 nike

正如标题所说,我想找出 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/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com