想练SDXL的LoRA,显存这关先掂量清楚

把SDXL的LoRA微调跑了一遍,顺手把显存记下来,给纠结要不要换卡的人参考。

条件:1024分辨率,FP16,开xformers。batch size压到1,占用十三G出头;batch拉到8,直接飙到十八G多。中间没别的花样,就改了个batch。

按这个数,想安稳跑SDXL的LoRA,16G是起步线,再往下就得各种妥协,动不动OOM。手里是老12G卡的别硬上,要么把分辨率降下来,要么老实batch 1慢慢磨。

12G老卡飘过,batch1加梯度检查点才勉强稳住

码住

所以16G是真起步,之前有人说12G随便跑纯属误导新人