diff --git a/requirements.txt b/requirements.txt
index 3f3776f..682ea65 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,9 +1,9 @@
 # Add here external Python modules dependencies, one per line. Module names
 # should match https://pypi.python.org/pypi names. For the full spec or
 # dependency lines, see https://pip.readthedocs.org/en/1.1/requirements.html
 vcversioner
 pygithub
 pyyaml
-tensorflow
+tensorflow-gpu
 keras
-h5py
\ No newline at end of file
+h5py