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 can i make it works when pytorch.dataloader num_workers > 0 with gpu_num > 1? #59

Open
hopkin-ghp opened this issue Mar 18, 2024 · 1 comment
Labels
help wanted Extra attention is needed

Comments

@hopkin-ghp
Copy link

Hello, David

GPURIR works on Pytorch online training with one GPU when I set the num_worker of dataloader is 0 through adding following code:

multiprocessing.set_start_method('spawn')
frezze_support

But it don't works when GPU num >1, do you have any good suggestions?

Thanks,
hopkin

@hopkin-ghp hopkin-ghp changed the title How can i make it works when pytorch.dataloader num_workers > 0 with gpu_num > 0? How can i make it works when pytorch.dataloader num_workers > 0 with gpu_num > 1? Mar 18, 2024
@DavidDiazGuerra
Copy link
Owner

Hello Hopkin,

Being able to use gpuRIR in a GPU different form the one used for training or even using it to perform different simulations in different GPUs would be indeed an interesting thing. However, unfortunately, I have no idea about how to do this and I guess it would need some (minor) changes in the library code.

I have no time to work on this right now, but if someone has some ideas about how to do it that would be an awesome contribution to the library.

Best,
David

@DavidDiazGuerra DavidDiazGuerra added the help wanted Extra attention is needed label Mar 23, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
help wanted Extra attention is needed
Projects
None yet
Development

No branches or pull requests

2 participants