diff --git a/requirements-gpu.txt b/requirements-gpu.txt
index 4df236e..aa15d8a 100644
--- a/requirements-gpu.txt
+++ b/requirements-gpu.txt
@@ -1,6 +1,6 @@
 flake8
 numpy
-tensorflow-gpu==1.15.2
+tensorflow-gpu==2.6.4
 tensorflow-hub==0.4.0
 bert-tensorflow==1.0.1
 click