google-deepmind / google-deepmind/alphafold
RTX PRO 6000 /SM_120 incompatibility
Open
- Dominant language
- Python
- Stars
- 14.9k
- Forks
- 2.9k
- PR merge metrics
- No merged PRs in 30d
Description
Hello,
is it possible to run AF2 on Blackwell cards? Would you be so kind as to provide dockerfile/requirements.txt ?
I tried a dozen of possible combinations between jax, cuda, haiku, etc., but none seem to work (XLA fails to initialise cuda, cannot see card, etc.)
thanks!
Contributor guide
Assessment
This issue has not been assessed yet.