Added install requirements to setup (#488)

* Added install requirements to setup * Update setup.py Co-authored-by: Aarni Koskela <[email protected]> --------- Co-authored-by: Aarni Koskela <[email protected]>
bitsandbytes-foundation · Jan 23, 2024 · 1e64210 · 1e64210
1 parent 407a8d3
commit 1e64210
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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=[