diff --git a/requirements.txt b/requirements.txt
index 0622644..d88c68e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -121,7 +121,7 @@ tensorboard-plugin-wit==1.6.0.post3
 tensorflow==2.2.0
 tensorflow-datasets==4.2.0
 tensorflow-estimator==2.2.0
-tensorflow-gpu==2.2.0
+tensorflow-gpu==2.4.0
 tensorflow-metadata==0.27.0
 tensorflow-probability==0.10.0
 termcolor==1.1.0