diff --git a/deeprole/code/nn_train/requirements.txt b/deeprole/code/nn_train/requirements.txt
index 65f9146..e235281 100644
--- a/deeprole/code/nn_train/requirements.txt
+++ b/deeprole/code/nn_train/requirements.txt
@@ -35,7 +35,7 @@ scipy==1.2.1
 simplegeneric==0.8.1
 six==1.12.0
 tensorboard==1.13.1
-tensorflow==1.13.1
+tensorflow==2.6.4
 tensorflow-estimator==1.13.0
 termcolor==1.1.0
 traitlets==4.3.2