diff --git a/requirements.txt b/requirements.txt
index 99b31ca..04d86e9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,7 +3,7 @@ opencv-python
 opencv-contrib-python
 wheel
 scikit-learn
-tensorflow-gpu==1.9.0
+tensorflow-gpu==2.6.4
 keras==2.2.1
 matplotlib
 prettytable