search

baker-laboratory / RoseTTAFold-All-Atom

Other
652 stars 118 forks source link

CUDA out of memory while running the 'nucleic_acid' example #22

Closed wjk1214 closed 8 months ago

wjk1214 commented 8 months ago

Thank you very much for providing us with such a useful tool. I have a question to ask: I successfully ran ‘Protein Monomers’ and 'protein_sm' using an RTX3060 with 12GB GPU memory. However, when I attempted to run ‘nucleic_acid’, I encountered a 'CUDA out of memory' error, despite the sequence not appearing particularly large. I'm unsure if this is normal? Thank you again for your contribution!

masterwhook commented 8 months ago

export this will be OK: export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb: 128

wjk1214 commented 8 months ago

export this will be OK: export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb: 128

Now it's working fine. Thank you very much for your generous response.