diff --git a/requirements.txt b/requirements.txt
index 2ee3df4..02504b1 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,5 +2,5 @@ absl-py
 easydict
 cython
 numpy
-tensorflow-gpu==1.14.0
+tensorflow-gpu==2.5.3
 tqdm