Jaxlib version error when building environment

[wangyuyy]

An error occurred when I tried the command pip install -r requirements.txt which said No matching distribution found for jaxlib==0.1.71+cuda111 Should I change the version of jaxlib? Appreciate a lot for your generous help.

[bonapark00]

Try pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html https://github.com/google/jax#installation