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

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: |