search

Avalon-Benchmark / avalon

A 3D video game environment and benchmark designed from scratch for reinforcement learning research
https://generallyintelligent.com/avalon/
GNU General Public License v3.0
181 stars 16 forks source link

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

Open sschertler opened 1 year ago

sschertler commented 1 year ago

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!

mx781 commented 1 year ago

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.