How (if possible) do I allocate more GPU utilization to Ray?

1

Hello,

I’m a beginner, I’m using Ray rllib to learn a PPO control policy. I’m making repeated calls to the Trainable.train() function.

I’m using the default config, and my GPU utilization is at about 20%. Is it possible to allocate more GPU memory to Ray to speed up training?

Thanks!