diff --git a/requirement.txt b/requirement.txt
index b367c3a..d03da41 100644
--- a/requirement.txt
+++ b/requirement.txt
@@ -2,5 +2,5 @@ numpy
 tqdm
 opencv-python
 keras==2.0.8
-tensorflow-gpu==1.4
+tensorflow-gpu==2.3.1
 scikit-learn
\ No newline at end of file