限制 Ray 的 CPU 资源

Question

我正在尝试管理我们用于日常任务（使用 Ray）的远程机器的资源。是否可以限制 Ray 使用的 CPU 数量（或等效的工作人员数量）？远程机器有 16 个内核，我可以限制 Ray 只使用其中的 12 个左右吗？提前致谢。

Answer 1

您可以使用以下方式限制资源：

# To start a head node.
ray start --head --num-cpus=12

# To start a non-head node.
ray start --redis-address=<redis-address> --num-cpus=12

或通过ray.init:

ray.init(num_cpus=12)

来源：ray documentation.