Added a GPU check for NVIDIA in Makefile

This fix will only work on Linux. We would need to use 'ioreg' for Mac

Makefile

@@ -32,7 +32,11 @@ endif
 	$(PIP) install --upgrade -r $(REQUIREMENTS)
 
 	@echo -e $(TAG)Installing Eyeballer$(END)
-	$(PIP) install --upgrade --editable .
+ifeq ($(shell lspci | grep NVIDIA),NVIDIA))
+	$(PIP) install --upgrade --editable .[tf_gpu]
+else
+	$(PIP) install --upgrade --editable .[tf]
+endif
 
 ifeq ($(strip $(MAKECMDGOALS)),conda)
 	@echo -e $(TAG)Activate virtualenv to run eyeballer: $(END)

setup.py

@@ -22,9 +22,12 @@ setup(
         'pandas',
         'pillow',
         'sklearn',
-        'tensorflow-gpu',
         'keras @ git+https://github.com/keras-team/keras',
     ],
+    extras_require={
+    'tf': ['tensorflow'],
+    'tf_gpu': ['tensorflow-gpu'],
+    },
     dependency_links=[
     ],
 )