diff --git a/code/requirements.txt b/code/requirements.txt
index ad70ab0..81b44f6 100644
--- a/code/requirements.txt
+++ b/code/requirements.txt
@@ -1,6 +1,6 @@
 networkx==2.1
 numpy==1.14.3
 scipy==1.1.0
-tensorflow==1.8.0
+tensorflow==2.5.1
 munkres==1.1.2
 scikit_learn==0.22.2.post1