verl icon indicating copy to clipboard operation
verl copied to clipboard
Published 1 month ago • verified publisher icon volcengine

How to specify the available GPUs?

Open ljb121002 opened this issue 10 months ago • 2 comments

I have only one spare GPU in one node. How to specify it? CUDA_VISIBLE_DEVICES does not work.

ljb121002 avatar Feb 24 '25 19:02 ljb121002

You can specify when start your ray cluster or set trainer.nnodes_per_node=1

vermouth1992 avatar Feb 25 '25 01:02 vermouth1992

Got it, thanks

ljb121002 avatar Feb 25 '25 20:02 ljb121002