manhattan_sdf icon indicating copy to clipboard operation
manhattan_sdf copied to clipboard

Published 20 hours ago • verified publisher icon zju3dv

为什么我运行strain_net.py代码的时候GPU内存极其容易爆满？

Open qinli-jian opened this issue 1 year ago • 1 comments

你好： 我在按照文件构建好项目，运行strain_net.py的时候，12G显存一下子就满了，我修改了batch_size也是容易爆满显存，请问你训练的机子都是显存很高的吗？ 谢谢！

qinli-jian avatar Aug 20 '22 11:08 qinli-jian

我们是用单张12G显卡训练的。你的pytorch版本是多少？建议使用1.6，参考：#8

ghy0324 avatar Aug 21 '22 06:08 ghy0324