diff --git a/requirements.txt b/requirements.txt
index 5a1f049..f838501 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 torch_geometric==1.3.2
 torch==1.2.0
-tensorflow==1.14.0
+tensorflow==2.6.4
 numpy==1.16.2
 networkx==2.3
 scipy==1.2.1