|
3 | 3 | FROM --platform=linux/amd64 ubuntu:22.04 as intermediate |
4 | 4 |
|
5 | 5 | RUN apt update && \ |
6 | | - DEBIAN_FRONTEND=noninteractive apt-get install -y build-essential gfortran git m4 wget cmake ninja-build \ |
| 6 | + DEBIAN_FRONTEND=noninteractive apt-get install -y build-essential gfortran git m4 wget \ |
7 | 7 | libopenblas-dev libfftw3-dev libhdf5-dev libhdf5-serial-dev libnetcdf-dev libnetcdff-dev libgl1-mesa-dev \ |
8 | 8 | python3-dev python3-pip python3-venv libxrender1 |
9 | 9 |
|
|
23 | 23 | cd .. && \ |
24 | 24 | rm -rf $mpich_prefix |
25 | 25 |
|
26 | | - |
27 | 26 | RUN /sbin/ldconfig |
28 | 27 |
|
| 28 | +RUN python3 -m pip install wheel |
| 29 | +RUN python3 -m venv /venv/ |
| 30 | + |
| 31 | +RUN /venv/bin/pip install -U pip |
| 32 | +# 2024-11-12 MJL: mpi4py < 4.0 in the next line is due to |
| 33 | +# https://github.com/hiddenSymmetries/simsopt/issues/455 |
| 34 | +# Once NERSC gets Shifter working with mpi4py >= 4 we can remove this inequality |
| 35 | +RUN /venv/bin/python -m pip install numpy scipy jax jaxlib f90nml "mpi4py<4.0" jupyter notebook ipython qsc sympy scikit-build ninja "pybind11[global]" cmake f90wrap h5py |
| 36 | +ENV PATH="/venv/bin:${PATH}" |
| 37 | + |
29 | 38 | RUN \ |
30 | 39 | git clone --depth 1 https://github.com/Reference-ScaLAPACK/scalapack.git scalapack && \ |
31 | 40 | mkdir scalapack/build && \ |
|
36 | 45 |
|
37 | 46 | RUN /sbin/ldconfig |
38 | 47 |
|
39 | | -RUN python3 -m pip install wheel |
40 | | -RUN python3 -m venv /venv/ |
41 | | - |
42 | | -RUN /venv/bin/pip install -U pip |
43 | | -# 2024-11-12 MJL: mpi4py < 4.0 in the next line is due to |
44 | | -# https://github.com/hiddenSymmetries/simsopt/issues/455 |
45 | | -# Once NERSC gets Shifter working with mpi4py >= 4 we can remove this inequality |
46 | | -RUN /venv/bin/python -m pip install numpy scipy jax jaxlib f90nml "mpi4py<4.0" jupyter notebook ipython qsc sympy scikit-build ninja "pybind11[global]" cmake f90wrap h5py |
47 | | -ENV PATH="/venv/bin:${PATH}" |
48 | 48 |
|
49 | 49 | RUN git clone --depth 1 https://github.com/PrincetonUniversity/SPEC.git /src/SPEC && \ |
50 | 50 | cd /src/SPEC && \ |
|
0 commit comments