Skip to content

Common installation errors

Roussel Des Nzoyem edited this page Jan 11, 2023 · 1 revision

Installing MPI4JAX can be tricky. You might find it helpful to consider the following steps when trying it in a Conda environment:

  • Make sure you have OpenMPI installed: on MAC, brew install open-mpi, or on Linux ...
  • If you prefer the MPICH implementation:
    • Don't install mpich-mpicc with Conda. If need be, uninstall it: conda uninstall -c conda-forge mpich-mpicc
    • Install MPICH system-wide (e.g. on Ubuntu): sudo apt install mpich
    • Remember to set the MPICC path to the system's: env MPICC=/usr/bin/mpicc
  • On Linux WSL, install x86_64-linux-gnu on Conda: conda install -c anaconda gcc_linux-64 (for some reason, MPI4JAX doesn't compile with the default)
  • Only then, you install and test mpi4jax

ENJOY :)

Clone this wiki locally