diff --git a/doc/requirements.txt b/doc/requirements.txt
index 909505d..5ea76ee 100644
--- a/doc/requirements.txt
+++ b/doc/requirements.txt
@@ -1,7 +1,7 @@
 numpy >= 1.17.4
 networkx >= 2.1
 scipy >= 1.1.0
-tensorflow == 1.14.0
+tensorflow == 2.5.1
 scikit-learn >= 0.22
 ogb_lite >= 0.0.3
 tqdm