diff --git a/Dockerfile b/Dockerfile
index f9eba822..da69e289 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -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
@@ -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