@@ -24,7 +24,12 @@ RUN apt-get update \
24
24
python3-pip python3-dev python3-venv\
25
25
# needed for openmp with clang
26
26
libomp-dev \
27
- doxygen
27
+ doxygen\
28
+ python3-pip python3-dev python3-venv \
29
+ && python3.12 -m venv .venv \
30
+ && .venv/bin/python -m pip install --no-cache-dir --upgrade\
31
+ pip pytest wheel setuptools cmakelang scipy numpy \
32
+ && rm -rf /var/lib/apt/lists/*
28
33
29
34
FROM debian_base AS debian_gcc_openmpi
30
35
@@ -44,7 +49,8 @@ RUN mkdir /tmp/openmpi-src \
44
49
&& CC=gcc CXX=g++ ./configure ${OPENMPI_CONFIGURE_OPTIONS} \
45
50
&& make all ${OPENMPI_MAKE_OPTIONS} \
46
51
&& make install \
47
- && cd && rm -rf /tmp/openmpi-src && ldconfig
52
+ && cd && rm -rf /tmp/openmpi-src && ldconfig \
53
+ && .venv/bin/python -m pip install --no-cache-dir mpi4py
48
54
49
55
# ### ADD DEFAULT USER ####
50
56
ARG USER=mpi
@@ -73,7 +79,8 @@ RUN mkdir /tmp/openmpi-src \
73
79
&& CC=clang CXX=clang++ ./configure ${OPENMPI_CONFIGURE_OPTIONS} \
74
80
&& make all ${OPENMPI_MAKE_OPTIONS} \
75
81
&& make install \
76
- && cd && rm -rf /tmp/openmpi-src && ldconfig
82
+ && cd && rm -rf /tmp/openmpi-src && ldconfig \
83
+ && .venv/bin/python -m pip install --no-cache-dir mpi4py
77
84
78
85
# ### ADD DEFAULT USER ####
79
86
ARG USER=mpi
@@ -102,7 +109,8 @@ RUN mkdir /tmp/mpich-src && cd /tmp/mpich-src \
102
109
&& CC=gcc CXX=g++ ./configure ${MPICH_CONFIGURE_OPTIONS} \
103
110
&& make ${MPICH_MAKE_OPTIONS} \
104
111
&& make install \
105
- && cd && rm -rf /tmp/mpich-src && ldconfig
112
+ && cd && rm -rf /tmp/mpich-src && ldconfig \
113
+ && .venv/bin/python -m pip install --no-cache-dir mpi4py
106
114
107
115
# ### ADD DEFAULT USER ####
108
116
ARG USER=mpi
@@ -131,7 +139,8 @@ RUN mkdir /tmp/mpich-src && cd /tmp/mpich-src \
131
139
&& CC=clang CXX=clang++ ./configure ${MPICH_CONFIGURE_OPTIONS} \
132
140
&& make ${MPICH_MAKE_OPTIONS} \
133
141
&& make install \
134
- && cd && rm -rf /tmp/mpich-src && ldconfig
142
+ && cd && rm -rf /tmp/mpich-src && ldconfig \
143
+ && .venv/bin/python -m pip install --no-cache-dir mpi4py
135
144
136
145
# ### ADD DEFAULT USER ####
137
146
ARG USER=mpi
0 commit comments