diff --git a/setup.py b/setup.py
index a71331dcf..b55678bf4 100644
--- a/setup.py
+++ b/setup.py
@@ -29,6 +29,8 @@ def read(fname):
     install_requires=['scipy'],
     packages=find_packages(),
     package_data={"": libs},
+    install_requires=['torch', 'numpy', 'scipy'],
+    extras_require={'benchmark': ['pandas', 'matplotlib']},
     long_description=read("README.md"),
     long_description_content_type="text/markdown",
     classifiers=[