DEV Community

Furkan Gözükara
Furkan Gözükara

Posted on

Multi-GPU FLUX Full Fine Tuning Experiments and Requirements on RunPod and Conclusions

I have done an extensive multi-GPU FLUX Full Fine Tuning / DreamBooth training experimentation on RunPod by using 2x A100–80 GB GPUs (PCIe) since this was commonly asked of me.

Image 1

  • Image 1 shows that only first part of installation of Kohya GUI took 30 minutes on a such powerful machine on a very expensive Secure Cloud pod — 3.28 USD per hour

  • There was also part 2, so just installation took super time

  • On Massed Compute, it would take like 2–3 minutes

  • This is why I suggest you to use Massed Compute over RunPod, RunPod machines have terrible hard disk speeds and they are like lottery to get good ones

Image 2, 3 and 4

  • Image 2 shows speed of our very best config FLUX Fine Tuning training shared below when doing 2x Multi GPU training

  • https://www.patreon.com/posts/kohya-flux-fine-112099700

  • Used config name is : Quality_1_27500MB_6_26_Second_IT.json

  • Image 3 shows VRAM usage of this config when doing 2x Multi GPU training

  • Image 4 shows the GPUs of the Pod

Image 5 and 6

  • Image 5 shows speed of our very best config FLUX Fine Tuning training shared below when doing a single GPU training

  • https://www.patreon.com/posts/kohya-flux-fine-112099700

  • Used config name is : Quality_1_27500MB_6_26_Second_IT.json

  • Image 6 shows this setup used VRAM amount

Image 7 and 8

  • Image 7 shows speed of our very best config FLUX Fine Tuning training shared below when doing a single GPU training and Gradient Checkpointing is disabled

  • https://www.patreon.com/posts/kohya-flux-fine-112099700

  • Used config name is : Quality_1_27500MB_6_26_Second_IT.json

  • Image 8 shows this setup used VRAM amount

Image 9 and 10

  • Image 9 shows speed of our very best config FLUX Fine Tuning training shared below when doing 2x Multi GPU training — this time Fused Backward Pass is disabled

  • https://www.patreon.com/posts/kohya-flux-fine-112099700

  • Used config name is : Quality_1_27500MB_6_26_Second_IT.json

  • Image 10 shows this setup used VRAM amount

Image 11

Conclusions

  • For multi-GPU FLUX Fine Tuning, you have to use at least 80 GB GPUs

  • When doing multi-GPU FLUX Fine Tuning, Fused Backward Pass brings 0 VRAM usage improvements but slows down the training — I am going to report this

  • With A100 GPU, you are able to reach 2.89 second / it — probably it will get better as you do more steps

  • With 2x A100 GPU, you are able to reach 4.1 second / it — effective speed 2.05 second / it

  • The speed gain is 0.75 / 2.9 = 26% — so 2x GPU training totally doesn’t worth it at the moment

  • If speed drop stays same due to multi-GPU overhead, 8x A100 may be beneficial but you have to experiment it properly and calculate speed gain

  • Currently single L40S would be way cheaper and faster

  • When doing multi-GPU FLUX LoRA training, we almost gain linear speed increase — I have tested with 8x RTX A6000 : https://www.patreon.com/posts/110879657

  • As shown in Image 11, there is also a chance that you will get a way worse performing pod

  • With same 2x A100 GPU and no visible difference, that random pod performed 1/4 speed of another same config pod — terrible terrible speed

 

 

 

 

 


 

Top comments (0)