Skip to content

Instantly share code, notes, and snippets.

@minrk

minrk/Dockerfile

Created November 30, 2024 13:49
Show Gist options
  • Save minrk/22707269668cdb7cc8933fc6d57e1d44 to your computer and use it in GitHub Desktop.
Save minrk/22707269668cdb7cc8933fc6d57e1d44 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Dockerfile
FROM quay.io/condaforge/miniforge3
RUN mamba create -n mpich -y -c minrk/label/libfabric-test 'mpich=4.2.3=libfab119*' mpi4py
# force install outdated libfabric
# can't request 1.14 via solver since 1.14 is not allowed by run_exports
# install via URL skips solve and just stages files
RUN mamba install -n mpich -y https://anaconda.org/minrk/libfabric/1.14.0/download/linux-64/libfabric-1.14.0-ha594dbc_0.conda
ENV MPICH_CH4_UCX_CAPABILITY_DEBUG=1 \
MPICH_CH4_OFI_CAPABILITY_DEBUG=1
CMD ["mamba", "run", "-n", "mpich", "sh", "-c", "mamba list --explicit; fi_info --version; mpiexec -n 4 python3 -c 'from mpi4py import MPI; print(MPI.COMM_WORLD.rank); MPI.COMM_WORLD.Barrier()'"]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment