Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
sschertler opened this issue Oct 13, 2023 · 1 comment
Open

Comments

@sschertler
Copy link

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
Copy link
Collaborator

mx781 commented Oct 16, 2023

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants