search

magenta / mt3

MT3: Multi-Task Multitrack Music Transcription
Apache License 2.0
1.43k stars 187 forks source link

colab notebook throws error in "Imports and Definitions" section #83

Closed tshimomura closed 2 years ago

tshimomura commented 2 years ago

It was working fine until a few days ago, But now, "Imports and Definitions" section says,

/usr/local/lib/python3.7/dist-packages/jax/_src/lib/__init__.py in check_jaxlib_version(jax_version, jaxlib_version, minimum_jaxlib_version)
     80            f'incompatible with jax version {jax_version}. Please '
     81            'update your jax and/or jaxlib packages.')
---> 82     raise RuntimeError(msg)
     83 
     84   return _jaxlib_version

RuntimeError: jaxlib version 0.3.20 is newer than and incompatible with jax version 0.3.15. Please update your jax and/or jaxlib packages.
GordemProo commented 2 years ago

And the previous cell outputs “Successfully installed jax-0.3.15” though.

playdasegunda commented 2 years ago

fix using:

!pip install --upgrade "jax[cuda]==0.3.15" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

GordemProo commented 2 years ago

*In the “Setup Environment” cell, replace the line !pip install jax==0.3.15 with the code in the comment above. Am I right, @lucasbr15? (Worked for me.)

iansimon commented 2 years ago

Should be fixed by #84, thank you @lucasbr15!