Skip to content

Commit

Permalink
[build] Add libhdf5-openmpi-dev in Dockerfile.
Browse files Browse the repository at this point in the history
In linux/arm64, openfermion requires h5py while h5py does not provide prebuilt
wheel so we need to compile it manually, which need libhdf5-openmpi-dev installed.
  • Loading branch information
hzhangxyz committed Sep 6, 2024
1 parent c5e5bb1 commit be65aa2
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
FROM python:3.12
RUN apt update
RUN apt install --yes libopenblas-dev libopenmpi-dev
RUN pip install build
RUN apt install --yes libopenblas-dev libopenmpi-dev libhdf5-openmpi-dev
RUN pip install build torch openfermion
COPY . TNSP
RUN python -m build TNSP/PyTAT -o dist -v
RUN python -m build TNSP/lazy_graph -o dist -v
Expand All @@ -10,4 +10,3 @@ RUN python -m build TNSP/tetragono -o dist -v
RUN python -m build TNSP/tetraku -o dist -v
RUN python -m build TNSP/tnsp_bridge -o dist -v
RUN pip install dist/*.whl
RUN pip install torch openfermion

0 comments on commit be65aa2

Please sign in to comment.