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,