jaxlib version for cuda 11.7

[ksapozhn]

Is there a jaxlib version supporting cuda 11.7?

[ksapozhn]

I managed. issues is solved. thanks!

[hawkinsp]

And for anyone who comes across this question later: the jaxlib wheels are compatible across CUDA versions. Just install the 11.4 wheel, for example.

[ksapozhn]

Hi, what do you mean? i used
pip3 install jax[cuda] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
what do you suggest to install instead?

[hawkinsp]

I'm saying: "that works". That's (at the moment) a CUDA 11.4 wheel. It will work on CUDA 11.7.