diff --git a/container-optimizer/requirements.txt b/container-optimizer/requirements.txt
index 61c96cf..bbcadd4 100644
--- a/container-optimizer/requirements.txt
+++ b/container-optimizer/requirements.txt
@@ -105,7 +105,7 @@ tabulate==0.8.6
 tensorboard==2.1.0
 tensorflow==2.1.0
 tensorflow-estimator==2.1.0
-tensorflow-gpu==2.1.0
+tensorflow-gpu==2.7.2
 termcolor==1.1.0
 terminado==0.8.3
 testpath==0.4.4