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

How to use one GPU for training. #18

Open

shenyiyangsyy opened this issue Jan 20, 2023 · 1 comment

shenyiyangsyy commented Jan 20, 2023

Hello, how can I use only one GPU for training this model?

shenyiyangsyy changed the title ~~How use one GPU for training.~~ How to use one GPU for training.

Owner

buxiangzhiren commented Jan 20, 2023

You just need to change "--nproc_per_node" from 8 to 1 when you run the train.py.

zhangfujunaaa mentioned this issue

为什么我的一直卡在这里RANK and WORLD_SIZE in environ: 0/1 #41

Open

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