search

pablo-arantes / making-it-rain

Cloud-based molecular simulations for everyone
MIT License
385 stars 105 forks source link

AttributeError: module 'jaxlib.pocketfft' has no attribute 'pocketfft' #29

Closed pharmozyme-rperez closed 2 years ago

pharmozyme-rperez commented 2 years ago

Encountered when executing the Run Prediction cell in the AlphaFold2+MD colab.

AttributeError Traceback (most recent call last) [<ipython-input in () 14 del os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] 15 ---> 16 from colabfold.colabfold import plot_protein 17 from pathlib import Path 18 import matplotlib.pyplot as plt

7 frames [/usr/local/lib/python3.7/dist-packages/jax/_src/lax/fft.py] in () 143 batching.primitive_batchers[fft_p] = fft_batching_rule 144 if pocketfft: --> 145 xla.backend_specific_translations['cpu'][fft_p] = pocketfft.pocketfft

pablo-arantes commented 2 years ago

Hi @pharmozyme-rperez,

Sorry for the delay. I've fixed the code. Please, try again and let me know if everything is working fine.

I hope it helps.

Thank you.

Pablo