diff --git a/ml_bench/requirements.txt b/ml_bench/requirements.txt
index 2c8d5db..fd7774c 100644
--- a/ml_bench/requirements.txt
+++ b/ml_bench/requirements.txt
@@ -1,4 +1,4 @@
 pyperf==2.0.0
 pyperformance==1.0.1
 absl-py==0.9.0
-tensorflow-gpu==2.2.0
+tensorflow-gpu==2.7.2