Colab not longer workin

[DHOFM]

This notebook worked until yesterday:

https://colab.research.google.com/github/magenta/mt3/blob/main/mt3/colab/music_transcription_with_transformers.ipynb

Since today it is dead now. I guess they changed python or a dependency. Will you maintain it or is the repo dead now?

[mattifrind]

The issue is that Google changed the colab cuda version from 11 to 12 so you get the following error:
CUDA backend failed to initialize

You can fix it by updating the version in the setup cell. Change this line:

!python3 -m pip install jax[cuda11_local] nest-asyncio pyfluidsynth==1.3.0 -e . -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

to this line:

!python3 -m pip install jax[cuda12_local] nest-asyncio pyfluidsynth==1.3.0 -e . -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

And it should work again.

[DHOFM]

Thanks,
I found this copy: https://colab.research.google.com/github/jyotidabass/Music-Transcription-with-Transformers/blob/main/Music_Transcription_with_Transformers.ipynb#scrollTo=-DiCjtDpyUMh

and it works. replaced jax[cuda11_local] with jax[cuda12_pip]==0.4.20 so gpu worked

[iansimon]

Fixed.