diff --git a/main/requirements.txt b/main/requirements.txt
index d4e84a8..f9e67df 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.6.4
 python_gflags==3.1.1
 python_magic==0.4.13