Installation#
As of PyMC v6, PyMC compiles models with the Numba backend by
default, so no extra system setup (compiler or BLAS) is required, and PyMC can
be installed with pip.
We also recommend installing nutpie, a fast NUTS sampler written in Rust. When nutpie is installed, PyMC selects it as the default NUTS sampler automatically — typically about 2x faster than PyMC’s built-in NUTS:
pip install "pymc[nutpie]"
To install PyMC on its own:
pip install pymc
Installing with conda#
As an alternative, PyMC can be installed with conda using
Anaconda or
Miniforge. We recommend a fresh
conda environment:
conda create -c conda-forge -n pymc_env pymc
conda activate pymc_env
If you like, replace the name pymc_env with whatever environment name you prefer.
To enable nutpie when using conda:
conda install -c conda-forge nutpie
See also
The conda-forge tips & tricks page to avoid installation issues when using multiple conda channels (e.g. defaults and conda-forge).
JAX sampling#
If you wish to enable sampling using the JAX backend via NumPyro, you need to install it manually as it is an optional dependency:
pip install numpyro
Similarly, to use the BlackJAX sampler instead:
pip install blackjax