Add fairscale and deepspeed back to the CI (#11147)

* Add fairscale and deepspeed back to the CI

* Add deepspeed to single GPU tests
This commit is contained in:
Lysandre Debut 2021-04-08 14:36:45 -04:00 committed by GitHub
parent 1ed24afe91
commit ba2cf5f90d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -34,6 +34,7 @@ jobs:
apt -y update && apt install -y libsndfile1-dev
pip install --upgrade pip
pip install .[sklearn,testing,onnxruntime,sentencepiece,speech]
pip install deepspeed
- name: Are GPUs recognized by our DL frameworks
run: |
@ -156,6 +157,8 @@ jobs:
apt -y update && apt install -y libsndfile1-dev
pip install --upgrade pip
pip install .[sklearn,testing,onnxruntime,sentencepiece,speech]
pip install fairscale
pip install deepspeed
- name: Are GPUs recognized by our DL frameworks
run: |