diff --git a/main/requirements.txt b/main/requirements.txt
index d4e84a8..3dfd1d0 100644
--- a/main/requirements.txt
+++ b/main/requirements.txt
@@ -2,7 +2,7 @@ setproctitle==1.1.10
 numpy==1.12.0
 pyglib==0.1
 scikit_image==0.9.3
-tensorflow_gpu==1.1.0
+tensorflow_gpu==2.7.2
 python_gflags==3.1.1
 python_magic==0.4.13