diff --git a/requirements.txt b/requirements.txt
index ceb5bd2..8b75d31 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -12,7 +12,7 @@ scikit-image>=0.14.1
 scikit-learn>=0.20.0
 scipy>=1.1.0
 seaborn>=0.9.0
-tensorflow-gpu==1.15.5
+tensorflow-gpu==2.5.1
 tensorflow-estimator==1.15.1
 tensorboard==1.15
 tensorflow-probability==0.8.0