From 43eefba143db94de2dc1a8b8fcf7f755aeb22291 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Thu, 8 Apr 2021 14:36:45 -0400 Subject: [PATCH] Add fairscale and deepspeed back to the CI (#11147) * Add fairscale and deepspeed back to the CI * Add deepspeed to single GPU tests --- .github/workflows/self-scheduled.yml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/.github/workflows/self-scheduled.yml b/.github/workflows/self-scheduled.yml index 3b72baea0d2b..c49a967d2aba 100644 --- a/.github/workflows/self-scheduled.yml +++ b/.github/workflows/self-scheduled.yml @@ -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: |