diff --git a/setup.py b/setup.py
index 106e033..9807b39 100644
--- a/setup.py
+++ b/setup.py
@@ -65,6 +65,7 @@ REQUIRED = [
     "lightgbm>=3.3.0",
     "psutil>=5.9.4",
     "torchvision>=0.15.1",
+    "tensorflow>=2.4.0",
 ]
 
 if get_platform() != MACOS: