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

关于模型训练问题 #9

Open

Hong-116 opened this issue Jun 21, 2024 · 1 comment

Hong-116 commented Jun 21, 2024

为什么我在训练模型时，cpu一直会往上升，最后会爬满，GPU反而基本上没用上，因为我没有多卡联机，我修改成了单机单gpu的代码，请问怎么解决

Owner

Jazzcharles commented Jul 4, 2024

目前的DDP应该是能够适配单GPU，可以尝试确认tensor和model的device是否是cuda?

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