diff --git a/docs/requirements.txt b/docs/requirements.txt
index 45d747a..c486b43 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -8,7 +8,7 @@ matplotlib
 numpy>=1.16
 pandas
 scipy>=0.17
-tensorflow==1.14
+tensorflow==2.3.1
 xport
 scikit-image
 requests
diff --git a/setup.py b/setup.py
index 52afcce..23d6dfd 100644
--- a/setup.py
+++ b/setup.py
@@ -24,7 +24,7 @@
             'cvxpy',
             'cvxopt',
             'Image',
-            'tensorflow==1.14',
+            'tensorflow==2.3.1',
 	    'h5py<3.0.0',
             'keras==2.3.1',
             'matplotlib',