Skip to content
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

多显卡训练的问题 #3

Open
xxxming730 opened this issue Oct 19, 2022 · 2 comments
Open

多显卡训练的问题 #3

xxxming730 opened this issue Oct 19, 2022 · 2 comments

Comments

@xxxming730
Copy link

您好,我现在正在用4张rtxa6000(48g)每张跑8个1024*1024的图像,但在all 阶段训练时就会出现显存溢出的问题
但是我将gpu_count改为1时,能够顺利完成训练,我不知道那里出了问题,我感到很困惑,希望您能够指导,万分感谢!

@RyanCCC
Copy link
Owner

RyanCCC commented Oct 20, 2022

这份代码不支持多卡训练的。如果你需要多卡训练可以查看原作者的开源的代码。好像他也给出了相关代码。

@xxxming730
Copy link
Author

这份代码不支持多卡训练的。如果你需要多卡训练可以查看原作者的开源的代码。好像他也给出了相关代码。

首先感谢您百忙之中抽出时间解答我的问题
谢谢您的提示,我会去源代码中进行查找,如果您有多卡训练的经验的话,还希望您能多多指点!
祝您身体健康,万事如意!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants