diff --git a/oss/Dockerfile b/oss/Dockerfile
index 095f03e..762ceee 100644
--- a/oss/Dockerfile
+++ b/oss/Dockerfile
@@ -21,10 +21,13 @@ RUN BAZEL_ARCH_SUFFIX="$(uname -m | sed s/aarch64/arm64/)" \
   && curl -sSL --fail -o /usr/local/bin/bazel "https://github.com/bazelbuild/bazel/releases/download/$BAZEL_VERSION/bazel-$BAZEL_VERSION-linux-$BAZEL_ARCH_SUFFIX" \
   && chmod a+x /usr/local/bin/bazel
 
+RUN apt-get install -y libhdf5-dev hdf5-tools
+
 # Install dependencies needed for array_record.
 RUN --mount=type=cache,target=/root/.cache \
   $PYTHON_BIN -m pip install -U \
     absl-py \
+    h5py \
     auditwheel \
     etils[epath] \
     patchelf \
@@ -32,4 +35,4 @@ RUN --mount=type=cache,target=/root/.cache \
     twine \
     wheel;
 
-WORKDIR "/tmp/array_record"
\ No newline at end of file
+WORKDIR "/tmp/array_record"