diff --git a/requirements.txt b/requirements.txt
index 1370878..abb85eb 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -24,7 +24,7 @@ scikit-learn>=0.20.0
 scikit-image>=0.13.1
 scipy>=1.0.0
 tensorboard>=1.12.0
-tensorflow-gpu==1.12.0
+tensorflow-gpu==2.5.1
 tensorpack>=0.9.0.1
 tqdm>=4.28.1
 yapf>=0.24.0