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

how to use multiple GPUs #36

Open
Eureka-Y opened this issue Apr 2, 2023 · 1 comment
Open

how to use multiple GPUs #36

Eureka-Y opened this issue Apr 2, 2023 · 1 comment

Comments

@Eureka-Y
Copy link

Eureka-Y commented Apr 2, 2023

Hi Eugen,
I want to use multiple GPUs to train the NADE model at the same time. I'm trying to write device_id=[0,1,2,3] or device_id =(0,1,2,3) in trainer.Trainer( ) ,but error message appears,I don't know how to write it correctly.

@EugenHotaj
Copy link
Owner

@Eureka-Y running on multiple GPUs is a bit involved and requires spawning multiple processes. The easiest way is to run the train.py script directly, or use it as an example to create your own training script.

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