towards solving #305

uclnlp · Oct 20, 2017 · 11428dc · 11428dc
1 parent 9b9c9a5
commit 11428dc
Showing 1 changed file with 1 addition and 0 deletions.
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,