多显卡训练的问题

[xxxming730]

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

[RyanCCC]

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

[xxxming730]

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

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