Closed ocstx closed 2 years ago
Sorry for the delay. It's fixed now via a pull request from @SarahBeecroft
I tried pip install --upgrade jax==0.2.14 jaxlib==0.1.69+cuda111 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html, but it doesn't work, There is still an error reported as “AttributeError: module 'jaxlib.xla_extension' has no attribute 'CompiledFunction'”
this line: pip install --upgrade jax==0.2.14 jaxlib==0.1.69+cuda111 -f https://storage.googleapis.com/jax-releases/jax_releases.html generates an error this issue was solved here: https://github.com/deepmind/alphafold/issues/510 with this: pip install --upgrade jax==0.2.14 jaxlib==0.1.69+cuda111 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
Shuld not be changed?