diff --git a/setup.py b/setup.py
index 16f9d9d..7259555 100644
--- a/setup.py
+++ b/setup.py
@@ -25,6 +25,7 @@
         'tensorflow-gpu==1.15.4',
         'numpy',
         'scipy',
+        'h5py',
         'pandas',
         'matplotlib',
         'seaborn',