issue with docker image installing jaxlib

[kellan]

output from build_docker.sh

#6 7.662 ERROR: Could not find a version that satisfies the requirement jaxlib==0.3.10+cuda11.cudnn82 (from jax[cuda]) (from versions: none) #6 7.735 ERROR: No matching distribution found for jaxlib==0.3.10+cuda11.cudnn82 (from jax[cuda]) Looks like this is an issue with the python 3.8 that is being installed by ubuntu and goes away with python 3.9+

[ryanalexmartin]

Duplicate issue

https://github.com/borisdayma/dalle-mini/issues/260

[MinahilRaza]

I changed https://storage.googleapis.com/jax-releases/jax_releases.html to https://storage.googleapis.com/jax-releases/jax_cuda_releases.html and it worked.

[farzanehnakhaee70]

Thanks @MinahilRaza I had the same problem and your solution solve it.