diff --git a/requirements.txt b/requirements.txt
index 5340dd5..3fd7a8c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,7 +9,7 @@ numpy
 pandas==0.24.1
 tqdm
 opencv-python
-tensorflow-gpu==1.14.0
+tensorflow-gpu==2.4.0
 tensorflow-hub==0.6.0
 matplotlib
 sklearn