Hi, I am unable to install the community-built version of Jax. Conda gives this error:
ERROR: Could not find a version that satisfies the requirement jaxlib==0.3.25 (from jax[cpu]) (from versions: 0.1.70, 0.1.70+cuda101, 0.1.70+cuda111, 0.1.71, 0.1.71+cuda101, 0.1.71+cuda111, 0.1.72, 0.1.72+cuda111, 0.1.73, 0.1.75, 0.1.75+cuda11.cudnn82, 0.1.76, 0.1.76+cuda11.cudnn82, 0.3.0, 0.3.0+cuda11.cudnn82, 0.3.2, 0.3.2+cuda11.cudnn82, 0.3.5, 0.3.5+cuda11.cudnn82, 0.3.7, 0.3.7+cuda11.cudnn82, 0.3.11, 0.3.11+cuda11.cudnn82, 0.3.14, 0.3.14+cuda11.cudnn82, 0.3.17, 0.3.17+cuda11.cudnn82, 0.3.20, 0.3.20+cuda11.cudnn82, 0.3.22, 0.3.22+cuda11.cudnn82, 0.3.24, 0.3.24+cuda11.cudnn82)
ERROR: No matching distribution found for jaxlib==0.3.25 (from jax[CPU])
Complete output:
(scvi-env) C:\Users\saini>pip install “jax[cpu]” -f https://whls.blob.core.windows.net/unstable/index.html --use-deprecated legacy-resolver
Looking in links: https://whls.blob.core.windows.net/unstable/index.html
Collecting jax[cpu]
Using cached jax-0.3.25.tar.gz (1.1 MB)
Collecting numpy>=1.20
Using cached numpy-1.23.5-cp39-cp39-win_amd64.whl (14.7 MB)
Collecting opt_einsum
Using cached opt_einsum-3.3.0-py3-none-any.whl (65 kB)
Collecting scipy>=1.5
Using cached scipy-1.9.3-cp39-cp39-win_amd64.whl (40.2 MB)
Collecting typing_extensions
Using cached typing_extensions-4.4.0-py3-none-any.whl (26 kB)
ERROR: Could not find a version that satisfies the requirement jaxlib==0.3.25 (from jax[cpu]) (from versions: 0.1.70, 0.1.70+cuda101, 0.1.70+cuda111, 0.1.71, 0.1.71+cuda101, 0.1.71+cuda111, 0.1.72, 0.1.72+cuda111, 0.1.73, 0.1.75, 0.1.75+cuda11.cudnn82, 0.1.76, 0.1.76+cuda11.cudnn82, 0.3.0, 0.3.0+cuda11.cudnn82, 0.3.2, 0.3.2+cuda11.cudnn82, 0.3.5, 0.3.5+cuda11.cudnn82, 0.3.7, 0.3.7+cuda11.cudnn82, 0.3.11, 0.3.11+cuda11.cudnn82, 0.3.14, 0.3.14+cuda11.cudnn82, 0.3.17, 0.3.17+cuda11.cudnn82, 0.3.20, 0.3.20+cuda11.cudnn82, 0.3.22, 0.3.22+cuda11.cudnn82, 0.3.24, 0.3.24+cuda11.cudnn82)
ERROR: No matching distribution found for jaxlib==0.3.25 (from jax[cpu])