google-deepmind / google-deepmind/alphafold
Running AlphaFold on models larger than AMD GPU available memory
- Dominant language
- Python
- Stars
- 14.9k
- Forks
- 2.9k
- PR merge metrics
- No merged PRs in 30d
Description
Dear developers,
I am facing an [out-of-memory issue](https://github.com/ROCm/ROCm/issues/3364#issuecomment-2216359771) (follow the link for a detailed log) when running ColabFold (which is a repackaging of AlphaFold) on AMD GPUs.
I have found [a comment on this repository](https://github.com/google-deepmind/alphafold/issues/340#issuecomment-1017699736) that suggests GPU managed memory can be used to leverage host memory together with GPU memory when the latter is not enough to hold the dataset. However, the same configuration does not work on AMD GPUs.
Do you have any suggestions on how to overcome the memory issue?
Thank you!
Contributor guide
Assessment
This issue has not been assessed yet.