google-deepmind / google-deepmind/alphafold

Running AlphaFold on models larger than AMD GPU available memory

Open
#958 4 comments 0 reactions 0 assignees View on GitHub
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

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.