baker-laboratory / RoseTTAFold-All-Atom

Other
614 stars 108 forks source link

Out of Memory error when using RFAA #79

Open AaranWang opened 5 months ago

AaranWang commented 5 months ago

The 48G GPU memory of the A40 still doesn't seem to be enough, is there a multi-GPU version of RFAA for multi-GPU cards?

ullahsamee commented 5 months ago

Check RoseTTaFold All-Atom (RFAA) - NVIDIA V100 vs A800 vs H800

RFAA

All three GPUs run into memory limitations when processing sequences longer than a certain length. -For V100, this limit is 800 a.a -For A800 and H800, is 1300 a.a