Update DeepSpeed version requirement in Dockerfile (#7326)

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2021-05-03 20:21:19 +02:00 · 2021-05-03 20:21:19 +02:00 · 7636d422fa
parent 39274273a4
commit 7636d422fa
1 changed files with 2 additions and 2 deletions
--- a/dockers/base-cuda/Dockerfile
+++ b/dockers/base-cuda/Dockerfile
@ -118,8 +118,8 @@ RUN \

 RUN \
    # install DeepSpeed
-    # TODO(@SeanNaren): 0.3.15 is broken - skipping to unblock
-    pip install 'deepspeed>=0.3.14,!=0.3.15'
+    # TODO(@SeanNaren): CI failing with `>=0.3.15` - skipping to unblock
+    pip install deepspeed==0.3.14

 RUN \
    # Show what we have