colab, kaggle notebook has a library dependency issue

[jkf87]

My code has been working fine for months in colab, but suddenly it won't run.
I got an error loading whisper-jax, saw the error message and tried changing the library version, but the problem still occurs in the dependencies.

!pip install -U "jax[cuda11_local]==0.4.13" "jaxlib[cuda11_local]==0.4.13" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
!pip install -q git+https://github.com/camenduru/whisper-jax.git datasets soundfile librosa yt_dlp cached_property

import jax
from whisper_jax import FlaxWhisperPipline
import jax.numpy as jnp

pipeline = FlaxWhisperPipline("openai/whisper-large-v2", dtype=jnp.float16)
from jax.experimental.compilation_cache import compilation_cache as cc
cc.initialize_cache("/content/jax_cache")

---

RuntimeError Traceback (most recent call last)
in <cell line: 6>()
4 get_ipython().system('pip install -q git+https://github.com/camenduru/whisper-jax.git datasets soundfile librosa yt_dlp cached_property')
5
----> 6 import jax
7 from whisper_jax import FlaxWhisperPipline
8 import jax.numpy as jnp

4 frames
/usr/local/lib/python3.10/dist-packages/jax/_src/lib/init.py in check_jaxlib_version(jax_version, jaxlib_version, minimum_jaxlib_version)
61 msg = (f'jaxlib is version {jaxlib_version}, but this version '
62 f'of jax requires version >= {minimum_jaxlib_version}.')
---> 63 raise RuntimeError(msg)
64
65 if _jaxlib_version > _jax_version:

RuntimeError: jaxlib is version 0.4.13, but this version of jax requires version >= 0.4.14.

[sanchit-gandhi]

Hey @jkf87! It looks like the error is coming from your JAX installation (mis-matching jax/jaxlib versions) - could you double check you have the correct installation here by referring to the official installation instructions? https://github.com/google/jax#installation