diff --git a/setup.py b/setup.py
index b883ae841d..d59e9f8cb1 100644
--- a/setup.py
+++ b/setup.py
@@ -29,14 +29,10 @@
 with io.open(readme_filename, encoding="utf-8") as readme_file:
     readme = readme_file.read()
 
-tensorboard_extra_require = [
-    "tensorflow-cpu>=2.3.0, <=2.5.0",
-    "grpcio~=1.38.1",
-    "six~=1.15.0",
-]
+tensorboard_extra_require = ["tensorflow >=2.3.0, <=2.5.0"]
 metadata_extra_require = ["pandas >= 1.0.0"]
 full_extra_require = tensorboard_extra_require + metadata_extra_require
-testing_extra_require = full_extra_require + ["grpcio-testing~=1.38.1"]
+testing_extra_require = full_extra_require + ["grpcio-testing"]
 
 
 setuptools.setup(