unable to correct CUDA vs. JAX version mismatch
Closed this issue · 1 comments
SJTUGuofei commented
CUDA backend failed to initialize: Found cuDNN version 8600, but JAX was built against version 8906, which is newer. The copy of cuDNN that is installed must be at least as new as the version against which JAX was built. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)
how can I deal with this issue?
syzhang092218-source commented
I guess you might want to update your CUDA and cuDNN versions.