restrict deepspeed version in CI (#8951)

2021-08-17 15:02:27 +02:00 · 2021-08-17 15:02:27 +02:00 · de22e40095
parent fdfd8644f2
commit de22e40095
1 changed files with 1 additions and 1 deletions
--- a/.azure-pipelines/gpu-tests.yml
+++ b/.azure-pipelines/gpu-tests.yml
@ -51,7 +51,7 @@ jobs:
    - bash: |
        python -c "fname = 'requirements/extra.txt' ; lines = [line for line in open(fname).readlines() if 'horovod' not in line] ; open(fname, 'w').writelines(lines)"
        pip install fairscale>=0.3.4
-        pip install "deepspeed>=0.4.3, !=0.4.4" # FIXME: bug with 0.4.4
+        pip install "deepspeed==0.4.3" # FIXME: bug with >= 0.4.4
        pip install . --requirement requirements/devel.txt
        pip list
      displayName: 'Install dependencies'