avalon icon indicating copy to clipboard operation
avalon copied to clipboard

Is it possible to run the PPO training script on an other GPU than 0?

Open sschertler opened this issue 1 year ago • 1 comments

I'd like to run multiple instances in parallel. Using export CUDA_VISIBLE_DEVICES=1 in the second session leads to an EOF error caused by the multithreading.connection library. Thanks in Advance!

sschertler avatar Oct 13 '23 06:10 sschertler

I'm unsure if we've tested this, but have you tried simply passing --train_gpu 1 to the training script? You might also want to change --inference_gpus, depending on what exactly you're doing.

mx781 avatar Oct 16 '23 08:10 mx781