限制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。