added optional requirements

jpauwels · Mar 30, 2015 · 040863f · 040863f
1 parent d84c8db
commit 040863f
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -13,11 +13,15 @@
       url='https://github.com/fchollet/keras',
       license='MIT',
       packages=[
-        'keras', 
-        'keras.layers', 
-        'keras.preprocessing', 
-        'keras.datasets', 
-        'keras.utils',
+          'keras',
+          'keras.layers',
+          'keras.preprocessing',
+          'keras.datasets',
+          'keras.utils',
       ],
-      install_requires=['numpy', 'scipy', 'theano']
+      install_requires=['numpy', 'scipy', 'theano'],
+      extras_require={
+          'images': ['pil'], # working with images
+          'CNNs': ['cudnn-python-wrappers'], # working with CNNs
+      }
 )