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

Multi GPU training speed #165

Open
kunazure opened this issue Oct 12, 2020 · 1 comment
Open

Multi GPU training speed #165

kunazure opened this issue Oct 12, 2020 · 1 comment

Comments

@kunazure
Copy link

Hello , I am using neural_sp to train aishell-1 and another task, but I find that when i use 4 GPU to train ASR model, it needs 8 hours per epoch, which is slow than I use single GPU,the time even grow when I add more GPU. Do you have this problem or there is something wrong with my envirment?
I use your tools to install the envirment, python=3.7.9 and pytorch=10.0 ,with cuda 10.1。

@hirofumi0810
Copy link
Owner

Hi, I think it's time to use DDP in pytorch. I'm still working on it internally. I'll update code soon.

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