diff --git a/setup.py b/setup.py
index 29f7a8466b..4227d0fd4e 100644
--- a/setup.py
+++ b/setup.py
@@ -154,7 +154,7 @@ def package_files(prefix: str, directory: str, extension: str):
 }
 
 extra_deps['deepspeed'] = [
-    'deepspeed==0.8.3',
+    'deepspeed==0.14.4',
     'pydantic>=1.0,<2',
 ]