使用 ray 和 torch 时的 CPU 利用率

Question

我在我的代码中使用射线和火炬，并为每个射线远程演员设置一个 CPU 核心来计算梯度（使用火炬包）。 但是我发现actor的CPU利用率可以在一段时间内达到300%，这似乎是不可能的，因为actor应该只使用一个CPU核心。 我想知道演员是否真的在使用更多的 CPU 资源，因为火炬可能会打开一个或多个线程来计算梯度。 我的操作系统是 Win10，CPU 是锐龙 5600H。 谢谢。

Answer 1

Ray 目前不会自动将 actor 固定到特定的 CPU 内核并阻止它使用其他 CPU 内核。 所以你所看到的是有道理的。

可以使用psutil之类的库将参与者固定到特定核心并阻止它使用其他核心。 如果您有许多并行任务/参与者，它们都是多线程的并且相互竞争资源（例如，因为它们使用 pytorch 或 numpy），这将很有帮助。