search

magenta / mt3

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

CUDA backend failed to initialize #145

Open mio-19 opened 8 months ago

mio-19 commented 8 months ago

I got the warning messsage on the "Load Model" stage. This might prevent the notebook to use gpu

WARNING:jax._src.xla_bridge:CUDA backend failed to initialize: Unable to load CUDA. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)
fengdor commented 8 months ago

i got the same warning... have you solved the problem now?

sandywu commented 8 months ago

replace jax[cuda11_local] wth jax[cuda12_pip]==0.4.20