-
Notifications
You must be signed in to change notification settings - Fork 7
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
train #6
Comments
How should I further set up the code to change to single gpu training |
Try this |
CUDA_VISIBLE_DEVICES=0 python -m torch.distributed.launch --nproc_per_node=1 --master_port 29506 train.py |
thanks |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
hello,i want to ask that I set multi-gpu training to single-gpu training,But report the following error:
and I set it like this in the source code:
The text was updated successfully, but these errors were encountered: