diff --git a/requirements.txt b/requirements.txt
index a2da57f..1c6048e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ absl-py
 easydict
 cython
 numpy==1.16.4
-tensorflow-gpu==1.14.0
+tensorflow-gpu==2.7.2
 tqdm
 msgpack
 msgpack_numpy