From 11428dc07aa1dd674bf74346dda0c0d40ea7daab Mon Sep 17 00:00:00 2001
From: Pasquale Minervini <pasquale.minervini@insight-centre.org>
Date: Fri, 20 Oct 2017 17:11:44 +0200
Subject: [PATCH] towards solving #305

---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index 87b8807c..5bcc7f98 100644
--- 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,