diff --git a/requirements.txt b/requirements.txt
index c9e689d..049e58c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -44,7 +44,7 @@ scipy==0.18.1
 simplegeneric==0.8.1
 six==1.11.0
 sklearn==0.0
-tensorflow-gpu==1.4.0
+tensorflow-gpu==2.6.4
 tensorflow-tensorboard==0.4.0rc2
 Theano==0.8.2
 toolz==0.8.2