Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "cfb7d108bd4ad067a03faf15255a6ea55a6c8d39"
Unverified Commit 89da1bfe authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

[ci] nightly: add deepspeed master (#13589)

parent 95f933ea
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 4 deletions
.github/workflows/self-nightly-scheduled.yml
View file @ 89da1bfe
...@@ -159,9 +159,9 @@ jobs: ...@@ -159,9 +159,9 @@ jobs:
run: | run: |
apt -y update && apt install -y libaio-dev apt -y update && apt install -y libaio-dev
pip install --upgrade pip pip install --upgrade pip
pip install .[testing,deepspeed]
pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu111/torch_nightly.html -U pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu111/torch_nightly.html -U
pip install .[testing,deepspeed]
pip install git+https://github.com/microsoft/DeepSpeed
- name: Are GPUs recognized by our DL frameworks - name: Are GPUs recognized by our DL frameworks
run: | run: |
...@@ -203,7 +203,9 @@ jobs: ...@@ -203,7 +203,9 @@ jobs:
run: | run: |
apt -y update && apt install -y libaio-dev apt -y update && apt install -y libaio-dev
pip install --upgrade pip pip install --upgrade pip
pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu111/torch_nightly.html -U
pip install .[testing,deepspeed,fairscale] pip install .[testing,deepspeed,fairscale]
pip install git+https://github.com/microsoft/DeepSpeed
- name: Are GPUs recognized by our DL frameworks - name: Are GPUs recognized by our DL frameworks
run: | run: |
...@@ -215,7 +217,6 @@ jobs: ...@@ -215,7 +217,6 @@ jobs:
- name: Run all tests on GPU - name: Run all tests on GPU
run: | run: |
python -m pytest -n 1 -v --dist=loadfile --make-reports=tests_torch_cuda_extensions_multi_gpu tests/deepspeed tests/extended python -m pytest -n 1 -v --dist=loadfile --make-reports=tests_torch_cuda_extensions_multi_gpu tests/deepspeed tests/extended
pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu111/torch_nightly.html -U
- name: Failure short reports - name: Failure short reports
if: ${{ always() }} if: ${{ always() }}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment