I am getting the above error after installing the latest version of jax and trying to import it for some computations:
!pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_releases.html
/usr/local/lib/python3.7/dist-packages/jax/_src/lax/linalg.py in <module>()
977 lu_pivots_to_permutation_p,
978 partial(_lu_pivots_to_permutation_gpu_lowering,
--> 979 gpu_linalg.cuda_lu_pivots_to_permutation),
980 platform='cuda')
981 mlir.register_lowering(
AttributeError: module 'jaxlib.cuda_linalg' has no attribute 'cuda_lu_pivots_to_permutation'
It would be helpful if someone could please look into this ASAP. This is a new error I received when running my code within the past hour and had no problem with this earlier.
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4