python - 限制 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 .