cryodrgn icon indicating copy to clipboard operation
cryodrgn copied to clipboard
verified publisher icon zhonge

Profile/reduce GPU memory usage for cryodrgn abinit_het

Open zhonge opened this issue 3 years ago • 0 comments

The default GPU memory usage for cryodrgn abinit_het maxes out at 40.5 GB when running on D=128 images with the default batch size of 8 (at the final fourier radius --l-end 32). Should try to keep this under 32GB or even 24GB.

zhonge avatar Jan 20 '23 16:01 zhonge