diff --git a/requirements.txt b/requirements.txt
index 7f89355..e22d021 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,7 +7,7 @@ ipdb==0.11
 scikit_image==0.14.0
 scipy==1.1.0
 numpy==1.14.5
-tensorflow_gpu==1.10.1
+tensorflow_gpu==2.7.2
 matplotlib==2.2.3
 quaternion==0.03.1
 PyYAML==5.1