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
180 stars 16 forks source link

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

Open sschertler opened 11 months ago

sschertler commented 11 months 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 11 months 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.