luwei0917 / DynamicBind

repo for DynamicBind: Predicting ligand-specific protein-ligand complex structure with a deep equivariant generative model
MIT License
202 stars 44 forks source link

GPU requirements #24

Closed MorganCThomas closed 5 months ago

MorganCThomas commented 5 months ago

What are the minimum GPU requirements to run this model?

Is an A100 necessary or can it be run on something with less memory?

Jnelen commented 5 months ago

For regular inference, a recent consumer-grade NVIDIA GPU should be sufficient. However, if you want to (re)train this type of model from scratch, an A100 GPU might be necessary.

MorganCThomas commented 5 months ago

Thanks!