Skip to content

Commit

Permalink
Try mpich 4.2
Browse files Browse the repository at this point in the history
  • Loading branch information
joezuntz committed Jul 12, 2024
1 parent acd23ed commit 9d7612f
Showing 1 changed file with 2 additions and 26 deletions.
28 changes: 2 additions & 26 deletions txpipe/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,30 +1,6 @@
FROM condaforge/mambaforge:23.3.1-1
FROM ghcr.io/lsstdesc/txpipe-base
MAINTAINER [email protected]

# Make, which we will need for everything
RUN apt-get update -y \
&& DEBIAN_FRONTEND="noninteractive" apt-get install -y make \
&& apt-get clean all

# We need a C compiler temporarily to install MPICH, which we want so that we use
# vendored MPI with conda. Not sure if we can get away with removing the gcc and g++
# compilers afterwards to save space but will try. We will use the conda gcc for everything
# else, though ideally everything would come through conda-forge.
# I have found that using the conda-forge supplied MPICH does not work
# with shifter on NERSC.
RUN apt-get update -y \
&& DEBIAN_FRONTEND="noninteractive" apt-get install -y gcc gfortran \
&& mkdir /opt/mpich \
&& cd /opt/mpich \
&& wget http://www.mpich.org/static/downloads/4.0.2/mpich-4.0.2.tar.gz \
&& tar xvzf mpich-4.0.2.tar.gz \
&& cd mpich-4.0.2 \
&& ./configure --disable-wrapper-rpath --disable-cxx --with-device=ch3 && make \
&& make install \
&& rm -rf /opt/mpich \
&& apt-get remove --purge -y gcc gfortran


# conda dependencies. As much as possible goes in here!
# should we be pinning versions?
RUN mamba install -c conda-forge -y scipy=1.8.0 \
Expand All @@ -34,7 +10,7 @@ RUN mamba install -c conda-forge -y scipy=1.8.0 \
"pandas=1.4.1" \
"dask=2023.5.0" \
scikit-learn \
'mpich=4.0.*=external_*' \
'mpich=4.2.*=external_*' \
'h5py=*=mpi_mpich_*'\
mpi4py \
"camb=1.5.*" \
Expand Down

0 comments on commit 9d7612f

Please sign in to comment.