[Fix] Could not find a version that satisfies the requirement jaxlib==0.4.25+cuda12.cudnn89

[BIEMAX]

Description

Fix the following error on MacOs:

ERROR: Could not find a version that satisfies the requirement jaxlib==0.4.25+cuda12.cudnn89; extra == "cuda12_pip" (from jax[cuda12-pip]) (from versions: 0.4.17, 0.4.18, 0.4.19, 0.4.20, 0.4.21, 0.4.22, 0.4.23, 0.4.24, 0.4.25)
ERROR: No matching distribution found for jaxlib==0.4.25+cuda12.cudnn89; extra == "cuda12_pip"

Reference

24 - Error "Could not find a version that satisfies the requirement jaxlib==0.4.25+cuda12.cudnn89"

[Bakobiibizo]

I got it installed manually putting this in the command line pip install jax[cuda12_pip] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

[BIEMAX]

@Bakobiibizo I did the same, but would be interesting if we could just run pip install instead installing this dependencies manually.

[StavrosD]

Nvidia no longer supports macOS since CUDA 11.7 (https://developer.nvidia.com/nvidia-cuda-toolkit-11_7_0-developer-tools-mac-hosts)

Jax does not support cuda on macOS so it is impossible to install it with cuda 12 on your computer.
https://jax.readthedocs.io/en/latest/installation.html