search

HannesStark / EquiBind

EquiBind: geometric deep learning for fast predictions of the 3D structure in which a small molecule binds to a protein
MIT License
477 stars 109 forks source link

Multiple GPUs #45

Closed Harry4FunResearch closed 2 years ago

Harry4FunResearch commented 2 years ago

Does the code support multiple GPUs?

HannesStark commented 2 years ago

No, but training only takes around 6GB of RAM.