diff --git a/setup.py b/setup.py
index e33b8d4..79312c2 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,8 @@
     'tensorboard',
     'tensorboardx',
     'merge_args',
-    'plotly'
+    'plotly',
+    'h5py'
 ]