Zuricho / ParallelFold

Modified version of Alphafold to divide CPU part (MSA and template searching) and GPU part. This can accelerate Alphafold when predicting multiple structures
https://parafold.sjtu.edu.cn
147 stars 45 forks source link

The GPU step is requiring large amounts of CPU VMEM #40

Open johnire80 opened 1 year ago

johnire80 commented 1 year ago

I can run the CPU step using modest amounts of resource - 16 core, 12G per core. But the GPU step requires 400G of VMEM. I am using a Nvidia A100 with 80G of GPU memory - and it is being used according to 'nvidia-smi'. The protein I am folding is very short - 197 amino acids. Is this normal? I would have hoped the GPU step would require much less CPU RAM.