diff --git a/hpvm/env.yaml b/hpvm/env.yaml
index b64c42e3f193df2ca7ac965ebd3dc57e5a8858be..c9cf79fc98bbcfca9c60f5234c7291bbe3fc534c 100644
--- a/hpvm/env.yaml
+++ b/hpvm/env.yaml
@@ -15,6 +15,7 @@ dependencies:
   - torchvision=0.8
   - tqdm=4.59
   - scipy==1.1.0
+  - h5py==2.10.0
   - pip:
     - argparse==1.4
     - onnx==1.8