gpt4 book ai didi

python - TensorFlow 是否将一台机器的所有 CPU 视为一个设备?

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

从我运行的实验来看,TensorFlow 似乎自动使用了一台机器上的所有 CPU。此外,似乎 TensorFlow 将所有 CPU 都称为/cpu:0。

我是对的吗,只有一台机器的不同 GPU 被索引并被视为单独的设备,但一台机器上的所有 CPU 都被视为单个设备?

从 TensorFlows 的角度来看,有没有什么方法可以让一台机器有多个 CPU 查看它?

最佳答案

默认情况下,进程可用的所有 CPU 都聚集在 cpu:0 设备下。

mrry 有答案here显示如何创建逻辑设备,如 /cpu:1/cpu:2

似乎没有将逻辑设备固定到特定物理内核或能够在 tensorflow 中使用 NUMA 节点的功能。

一种可能的解决方法是在一台机器上使用分布式 TensorFlow 和多个进程,并在 Linux 上使用 taskset 将特定进程固定到特定内核

关于python - TensorFlow 是否将一台机器的所有 CPU 视为一个设备?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/38836269/

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