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