diff --git a/requirements.txt b/requirements.txt
index 7bdce5c..3a55778 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,4 +2,4 @@ networkx==2.2
 scipy==1.1.0
 setuptools==40.6.3
 numpy==1.15.4
-tensorflow==1.15.2
+tensorflow==2.3.1
diff --git a/setup.py b/setup.py
index 9b0ba90..66c5daf 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@
       download_url='https://github.com/tkipf/gcn',
       license='MIT',
       install_requires=['numpy>=1.15.4',
-                        'tensorflow>=1.15.2,<2.0',
+                        'tensorflow>=1.15.2,<3.0',
                         'networkx>=2.2',
                         'scipy>=1.1.0'
                         ],