-
Notifications
You must be signed in to change notification settings - Fork 7.4k
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
Help with multi-GPU training #2442
Comments
Please see the example https://github.com/facebookresearch/detectron2/blob/master/tools/train_net.py on how to implement multi-gpu training. |
Thanks, |
I tried: but I am getting an error: train_net.py: error: unrecognized arguments: --local_rank=0 it seems the torch is adding a --local_rank to the args. |
Instructions To Reproduce the Issue:
I am trying to use multi-GPU training using Jupiter within DLVM (google compute engine with 4 Tesla T4).
my code only runs on 1 GPU, the other 3 are not utilized.
I am able to train with custom dataset and getting acceptable results, but wish to use 4 GPUs for faster training.
The text was updated successfully, but these errors were encountered: