diff --git a/requirements.txt b/requirements.txt
index 1e53145..4d06b22 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -65,7 +65,7 @@ sklearn==0.0
 smmap2==2.0.3
 snakeviz==0.4.2
 system-service==0.3
-tensorflow-gpu==1.4.0
+tensorflow-gpu==2.7.2
 tensorflow-tensorboard==0.4.0rc3
 tornado==4.5.2
 ubuntu-drivers-common==0.0.0