diff --git a/requirements.txt b/requirements.txt
index 114ec91..0e6689e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 metis
 torch==1.4.0
-tensorflow==2.0.1
+tensorflow==2.6.4
 tensorboard==2.0.2
 networkx==2.4
 scipy>=1.4.1