diff --git a/docs/requirements.txt b/docs/requirements.txt
index 1be4caf6d..13384844f 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -14,7 +14,7 @@ sphinx
 sphinx_intl
 sphinx_markdown_tables
 sphinx_rtd_theme
-tb-nightly
+tensorboard
 torch>=1.7
 torchvision
 tqdm
diff --git a/requirements.txt b/requirements.txt
index 82437c2b6..45ce62034 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,7 +8,7 @@ pyyaml
 requests
 scikit-image
 scipy
-tb-nightly
+tensorboard
 torch>=1.7
 torchvision
 tqdm