diff --git a/setup.py b/setup.py
index 3059bd9e..7fd5d5ac 100644
--- a/setup.py
+++ b/setup.py
@@ -91,7 +91,7 @@ def get_kernels_whl_url(
     "tokenizers>=0.12.1",
     "typing_extensions>=4.8.0",
     "accelerate",
-    "datasets",
+    "datasets>=2.20",
     "zstandard",
 ]