diff --git a/docs/requirements.txt b/docs/requirements.txt
index 7bd3e8a..6828813 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -1,6 +1,6 @@
 numpy==1.15.1
 Pillow==8.2.0
 scipy==1.1.0
-tensorflow-gpu==1.11.0
+tensorflow-gpu==2.4.2
 wget==3.2
 seaborn==0.9.0