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

是否支持多gpu训练 #4

Closed

sillytheif opened this issue Jul 13, 2023 · 1 comment

sillytheif commented Jul 13, 2023

想请教一下，这个代码支持多gpu一起训嘛

Collaborator

hi-zhenyu commented Sep 14, 2023

目前代码并不支持多GPU运行，NCR单张消费级显卡应该就能运行，且时间不会太长，因此未来暂无支持多GPU的计划。

如果确有这方面的需求的话，建议方式是是直接使用DataParallel进行单机多GPU运行，但这类实现运行效率不高。如果有DDP相关使用经验的话，更推荐DDP实现。也欢迎支持多GPU的PR。

hi-zhenyu closed this as completed

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