Conda inconsistency: jax and numpyro


I get this message from Conda :

The following packages are causing the inconsistency:

  • conda-forge/noarch::numpyro==0.8.0=pyhd8ed1ab_0
  • conda-forge/noarch::jax==0.2.24=pyhd8ed1ab_0

Here is how I have proceeded for installation on Linux few days ago

conda create -n jaxcosmo python=3.8
conda activate jaxcosmo
conda install -c conda-forge numpyro
pip install --upgrade “jax[cuda]” -f

An idea?

Look like jax comes from conda-forge. You might uninstall jax first before using pip. Or you can install jax first, then using conda to install numpyro.

Ok I will try next. Thanks