Merge pull request #3 from TimDettmers/cpuonly

Add a CPU-only build option

1 files changed, 11 insertions, 11 deletions

diff --git a/setup.py b/setup.py
index 59cd78e..2402c02 100644
--- a/setup.py
+++ b/setup.py
@@ -6,27 +6,27 @@ import os
 from setuptools import setup, find_packages
 
 
-
 def read(fname):
     return open(os.path.join(os.path.dirname(__file__), fname)).read()
 
 
+version = os.getenv("CUDA_VERSION", "cpu")
+
 setup(
-    name = f"bitsandbytes-cuda{os.environ['CUDA_VERSION']}",
-    version = "0.26.0",
-    author = "Tim Dettmers",
-    author_email = "dettmers@cs.washington.edu",
-    description = ("8-bit optimizers and quantization routines."),
-    license = "MIT",
-    keywords = "gpu optimizers optimization 8-bit quantization compression",
-    url = "http://packages.python.org/bitsandbytes",
+    name="bitsandbytes",
+    version=f"0.26.0+{version}",
+    author="Tim Dettmers",
+    author_email="dettmers@cs.washington.edu",
+    description="8-bit optimizers and quantization routines.",
+    license="MIT",
+    keywords="gpu optimizers optimization 8-bit quantization compression",
+    url="http://packages.python.org/bitsandbytes",
     packages=find_packages(),
     package_data={'': ['libbitsandbytes.so']},
     long_description=read('README.md'),
-    long_description_content_type = 'text/markdown',
+    long_description_content_type='text/markdown',
     classifiers=[
         "Development Status :: 4 - Beta",
         'Topic :: Scientific/Engineering :: Artificial Intelligence'
     ],
 )
-