Unverified Commit d1fd64e7 authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

clear ~/.cache/torch_extensions between builds (#14520)

parent 3772af49
...@@ -205,8 +205,9 @@ jobs: ...@@ -205,8 +205,9 @@ jobs:
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 --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu111/torch_nightly.html -U
pip install .[testing,deepspeed,fairscale] rm -rf ~/.cache/torch_extensions/ # shared between conflicting builds
pip install git+https://github.com/microsoft/DeepSpeed pip install .[testing,fairscale]
pip install git+https://github.com/microsoft/DeepSpeed # testing bleeding edge
- name: Are GPUs recognized by our DL frameworks - name: Are GPUs recognized by our DL frameworks
run: | run: |
......
...@@ -437,6 +437,7 @@ jobs: ...@@ -437,6 +437,7 @@ 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
rm -rf ~/.cache/torch_extensions/ # shared between conflicting builds
pip install .[testing,deepspeed,fairscale] pip install .[testing,deepspeed,fairscale]
- name: Are GPUs recognized by our DL frameworks - name: Are GPUs recognized by our DL frameworks
......
...@@ -429,6 +429,7 @@ jobs: ...@@ -429,6 +429,7 @@ 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
rm -rf ~/.cache/torch_extensions/ # shared between conflicting builds
pip install .[testing,deepspeed,fairscale] pip install .[testing,deepspeed,fairscale]
- name: Are GPUs recognized by our DL frameworks - name: Are GPUs recognized by our DL frameworks
......
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