diff --git a/requirements.txt b/requirements.txt
index ff6e624..fa4bd89 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -15,7 +15,7 @@ requests==2.18.4
 scipy==0.19.1
 six==1.10.0
 smart-open==1.5.3
-tensorflow-gpu==1.2.1
+tensorflow-gpu==2.6.4
 urllib3==1.22
 Werkzeug==0.12.2
 ciseau==1.0.0