diff --git a/code/requirements.txt b/code/requirements.txt
index 36fba38..a62a741 100644
--- a/code/requirements.txt
+++ b/code/requirements.txt
@@ -61,7 +61,7 @@ subprocess32==3.5.3
 tensorboard==1.8.0
 tensorflow==1.8.0
 tensorflow-estimator==1.13.0
-tensorflow-gpu==1.8.0
+tensorflow-gpu==2.5.3
 termcolor==1.1.0
 tqdm==4.31.1
 Werkzeug==0.15.2