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.

Sign up for GitHub

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

Jump to bottom

Fitting hangs at "cleaning up ddp environment..." when tpu_cores=8 #6390

Closed
KijitoraButi opened this issue Mar 7, 2021 · 4 comments · Fixed by #6816
Closed

Fitting hangs at "cleaning up ddp environment..." when tpu_cores=8 #6390

KijitoraButi opened this issue Mar 7, 2021 · 4 comments · Fixed by #6816
Assignees
@kaushikb11
Labels
accelerator: tpu Tensor Processing Unit bug Something isn't working help wanted Open to be worked on priority: 0 High priority task

Comments

@KijitoraButi
Copy link

🐛 Bug

When setting tpu_cores of Trainer to 8, fitting hangs at "cleaning up ddp environment...".

Please reproduce using the BoringModel

https://colab.research.google.com/drive/1tJswNaT0I-GrGsi6ngwwRDUmeFY1pFr3?usp=sharing

To Reproduce

Run above URL notebook.

Expected behavior

Trainer.fit ends normally .

Environment

  • PyTorch Version (e.g., 1.0): 1.7.0a0+7e71a98
  • OS (e.g., Linux): Linux
  • How you installed PyTorch (conda, pip, source): pip
  • Build command you used (if compiling from source): N/A
  • Python version: 3.7.10
  • CUDA/cuDNN version: N/A
  • GPU models and configuration: N/A
  • Any other relevant information: I tried the notebook on Google Colab with TPU.
@KijitoraButi KijitoraButi added bug Something isn't working help wanted Open to be worked on labels Mar 7, 2021
@awaelchli awaelchli added accelerator: tpu Tensor Processing Unit with code labels Mar 7, 2021
@edenlightning edenlightning added the priority: 0 High priority task label Mar 8, 2021
@edenlightning edenlightning assigned kaushikb11 and unassigned tchaton Mar 22, 2021
@kaushikb11
Copy link
Contributor

Hi @KijitoraButi, thank you for opening the issue! Yes, tried to reproduce the issue & it does hang. We will look into fixing it soon.

This was referenced Apr 3, 2021
Merged
Merged
@KijitoraButi
Copy link
Author

I was able to train the model properly.
Thanks for the fix!

@kaushikb11
Copy link
Contributor

@KijitoraButi Awesome! Did you try the master?

@KijitoraButi
Copy link
Author

@kaushikb11
Yes!
I updated Lightning to Master and then ran the code and it worked fine.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@kaushikb11 kaushikb11

Labels
accelerator: tpu Tensor Processing Unit bug Something isn't working help wanted Open to be worked on priority: 0 High priority task
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

[Fix] TPU Training Type Plugin kaushikb11/pytorch-lightning
5 participants
@awaelchli @tchaton @kaushikb11 @edenlightning @KijitoraButi