diff --git a/requirements.txt b/requirements.txt
index 0bb31d362..52f858faf 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -24,5 +24,5 @@ flake8==3.8.1
 isort==5.10.1
 black==21.4b2
 autoflake
-tensorboardX<=2.5.1
+tensorboardX==2.6
 pytest
diff --git a/setup.py b/setup.py
index 5874efab7..5d94c945a 100644
--- a/setup.py
+++ b/setup.py
@@ -133,7 +133,7 @@ def get_libai_configs() -> List[str]:
             "isort==5.10.1",
             "black==21.4b ",
             "autoflake",
-            "tensorboardX<=2.5.1",
+            "tensorboardX==2.6",
             "pytest",
         ],
         packages=find_packages(),