towards solving #305

Pasquale Minervini · Pasquale Minervini · commit 11428dc07aa1 · 2017-10-20T17:11:44.000+02:00
diff --git a/setup.py b/setup.py
@@ -52,6 +52,7 @@ def run(self):
       install_requires=requirements + tests_requirements,
       extras_require={
           'tests': tests_requirements,
+          'gpu': ['tensorflow-gpu>=1.3']
       },
       setup_requires=['pytest-runner'] + requirements,
       tests_require=requirements,
