pb with conda version 0.8.0

[jecampagne]

Hello I've done fresh install conda install -c conda-forge numpyro then

(jaxcosmo) cca010:users/campagne/nested_sampling$ conda list
# packages in environment at /sps/lsst/users/campagne/anaconda3/envs/jaxcosmo:
#
# Name                    Version                   Build  Channel
_libgcc_mutex             0.1                 conda_forge    conda-forge
_openmp_mutex             4.5                       1_gnu    conda-forge
absl-py                   0.13.0           py38h06a4308_0  
arviz                     0.11.2                   pypi_0    pypi
binutils_impl_linux-64    2.33.1               he6710b0_7  
binutils_linux-64         2.33.1              h9595d00_15  
blas                      1.0                    openblas  
brotli                    1.0.9                he6710b0_2  
ca-certificates           2021.5.30            ha878542_0    conda-forge
camb                      1.1.0            py38h6e990d7_3    conda-forge
certifi                   2021.5.30        py38h578d9bd_0    conda-forge
cftime                    1.5.1                    pypi_0    pypi
colorama                  0.4.4              pyh9f0ad1d_0    conda-forge
corner                    2.2.1                    pypi_0    pypi
cycler                    0.10.0                   py38_0  
fast-pt                   3.1.0              pyh9f0ad1d_0    conda-forge
fftw                      3.3.9                h27cfd23_1  
fonttools                 4.25.0             pyhd3eb1b0_0  
freetype                  2.10.4               h5ab3b9f_0  
gcc_impl_linux-64         7.3.0                habb00fd_1  
gcc_linux-64              7.3.0               h553295d_15  
gfortran_impl_linux-64    7.3.0                hdf63c60_1  
gfortran_linux-64         7.3.0               h553295d_15  
gmp                       6.2.1                h2531618_2  
gmpy2                     2.0.8            py38hd5f6e3b_3  
gsl                       2.7                  he838d99_0    conda-forge
jax                       0.2.21             pyhd8ed1ab_0    conda-forge
jaxlib                    0.1.71           py38h43a58ef_0    conda-forge
jaxns                     0.0.7                    pypi_0    pypi
jpeg                      9d                   h7f8727e_0  
kiwisolver                1.3.1            py38h2531618_0  
lcms2                     2.12                 h3be6417_0  
ld_impl_linux-64          2.33.1               h53a641e_7  
libblas                   3.9.0                8_openblas    conda-forge
libcblas                  3.9.0                8_openblas    conda-forge
libffi                    3.3                  he6710b0_2  
libgcc-ng                 11.1.0               hc902ee8_8    conda-forge
libgfortran-ng            7.5.0               ha8ba4b0_17  
libgfortran4              7.5.0               ha8ba4b0_17  
libgomp                   11.1.0               hc902ee8_8    conda-forge
libopenblas               0.3.12          pthreads_hb3c22a3_1    conda-forge
libpng                    1.6.37               hbc83047_0  
libstdcxx-ng              11.1.0               h56837e0_8    conda-forge
libtiff                   4.2.0                h85742a9_0  
libwebp-base              1.2.0                h27cfd23_0  
lz4-c                     1.9.3                h295c915_1  
matplotlib-base           3.4.2            py38hab158f2_0  
mock                      4.0.3              pyhd3eb1b0_0  
mpc                       1.1.0                h10f8cd9_1  
mpfr                      4.0.2                hb69a4c5_1  
mpmath                    1.2.1            py38h06a4308_0  
munkres                   1.1.4                      py_0  
ncurses                   6.2                  he6710b0_1  
netcdf4                   1.5.7                    pypi_0    pypi
numpy                     1.20.3           py38h3dbb7de_0  
numpy-base                1.20.3           py38h39b7dee_0  
numpyro                   0.8.0              pyhd8ed1ab_0    conda-forge
olefile                   0.46               pyhd3eb1b0_0  
openjpeg                  2.4.0                h3ad879b_0  
openssl                   1.1.1l               h7f98852_0    conda-forge
opt_einsum                3.3.0              pyhd8ed1ab_1    conda-forge
pandas                    1.1.3            py38he6710b0_0    anaconda
patsy                     0.5.1                    py38_0  
pillow                    8.3.1            py38h2c7a002_0  
pip                       21.2.2           py38h06a4308_0  
pyccl                     2.2.0            py38h93ac01b_2    conda-forge
pyparsing                 2.4.7              pyhd3eb1b0_0  
python                    3.8.11          h12debd9_0_cpython  
python-dateutil           2.8.2              pyhd3eb1b0_0  
python-flatbuffers        1.12               pyhd3eb1b0_0  
python_abi                3.8                      2_cp38    conda-forge
pytz                      2021.1             pyhd3eb1b0_0  
pyyaml                    5.4.1            py38h27cfd23_1  
readline                  8.1                  h27cfd23_0  
scipy                     1.7.1            py38hc65b3f8_2  
seaborn                   0.11.2               hd8ed1ab_0    conda-forge
seaborn-base              0.11.2             pyhd8ed1ab_0    conda-forge
setuptools                58.0.4           py38h06a4308_0  
six                       1.16.0             pyhd3eb1b0_0  
sqlite                    3.36.0               hc218d9a_0  
statsmodels               0.12.2           py38h27cfd23_0  
sympy                     1.8              py38h06a4308_0  
tk                        8.6.11               h1ccaba5_0  
tornado                   6.1              py38h27cfd23_0  
tqdm                      4.62.3             pyhd8ed1ab_0    conda-forge
wheel                     0.37.0             pyhd3eb1b0_1  
xarray                    0.19.0                   pypi_0    pypi
xz                        5.2.5                h7b6447c_0  
yaml                      0.2.5                h7b6447c_0  
zlib                      1.2.11               h7b6447c_3  
zstd                      1.4.9                haebb681_0  

One can see numpyroversion seems 0.8.0; now

(jaxcosmo) cca010:users/campagne/nested_sampling$ python
Python 3.8.11 (default, Aug  3 2021, 15:09:35) 
[GCC 7.5.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import numpyro

gives 0.7.2!!!

Where is the pb?

Thanks (nb. I was tying to run Example: Nested Sampling for Gaussian Shells)

[fehiepsi]

Could you try to do pip uninstall numpyro 2 times then install it again?

[jecampagne]

Well now even after conda reinstall

>>> import numpyro
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'numpyro'

while

(jaxcosmo) cca010:users/campagne/nested_sampling$ conda list | grep numpyro
numpyro                   0.8.0              pyhd8ed1ab_0    conda-forge

[fehiepsi]

I just created a fresh conda environment and the installation is working as expected. You might want to create your environment again.

[jecampagne]

Ok let me try along this line. Thanks

[jecampagne]

Ouf. should be ok now. Thanks.

By the way, how do you install jaxns in the conda environment.