add deepspeed fairscale (#10116)
This commit is contained in:
3
.github/workflows/self-scheduled.yml
vendored
3
.github/workflows/self-scheduled.yml
vendored
@@ -221,6 +221,8 @@ jobs:
|
|||||||
pip install --upgrade pip
|
pip install --upgrade pip
|
||||||
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece]
|
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece]
|
||||||
pip install git+https://github.com/huggingface/datasets
|
pip install git+https://github.com/huggingface/datasets
|
||||||
|
pip install fairscale
|
||||||
|
pip install deepspeed
|
||||||
pip list
|
pip list
|
||||||
|
|
||||||
- name: Are GPUs recognized by our DL frameworks
|
- name: Are GPUs recognized by our DL frameworks
|
||||||
@@ -353,4 +355,3 @@ jobs:
|
|||||||
with:
|
with:
|
||||||
name: run_all_tests_tf_multi_gpu_test_reports
|
name: run_all_tests_tf_multi_gpu_test_reports
|
||||||
path: reports
|
path: reports
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user