New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # 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

# for free to join this conversation on GitHub. Already have an account? # to comment