Limiting CPU resources of Ray

user_12637530 0
Viewed 423

I'm trying to manage the resources of a remote machine that we use for a daily task (that uses Ray). Is it possible to limit the number of CPUs (or equivalently the number of workers) that Ray uses?

The remote machine has 16 cores. Can I limit Ray to use only 12 of them or so?

ray distributed-computing cpu python
1 Answers

You can limit resources using:

# 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

Or via ray.init:

ray.init(num_cpus=12)

Source: Ray documentation.

user_141860870
Related