Open mio-19 opened 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.)
i got the same warning... have you solved the problem now?
replace jax[cuda11_local] wth jax[cuda12_pip]==0.4.20
jax[cuda11_local]
jax[cuda12_pip]==0.4.20
I got the warning messsage on the "Load Model" stage. This might prevent the notebook to use gpu