diff --git a/setup.py b/setup.py index b683bfcf1..7b4c0e103 100644 --- a/setup.py +++ b/setup.py @@ -27,6 +27,8 @@ def read(fname): url="https://github.com/TimDettmers/bitsandbytes", 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=[