Update tf requirements

Irina Nicolae · Irina Nicolae · commit 48d9bf1dec20 · 2018-04-11T12:17:22.000+01:00
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-tensorflow==1.1
+tensorflow==1.5
 h5py
 Keras
 matplotlib
diff --git a/setup.py b/setup.py
@@ -30,8 +30,11 @@
                         'Keras',
                         'scipy',
                         'matplotlib',
-                        'tensorflow',
                         'setuptools'],
+      extras_require={
+        "tf": ["tensorflow<=1.5.0"],
+        "tf_gpu": ["tensorflow-gpu<=1.5.0"],
+      },
       # extras_require={
           # 'tests': ['pytest',
           #           'pytest-pep8',
