Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature] Move 'Install MMCV' to a independent CI item. #602

Merged
merged 8 commits into from
Jun 17, 2021
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
24 changes: 20 additions & 4 deletions .github/workflows/build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -37,19 +37,26 @@ jobs:
include:
- torch: 1.3.0+cpu
torchvision: 0.4.1+cpu
torch_version: 1.3.0
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We may add device to specify cuda / cpu.

- torch: 1.5.0+cpu
torchvision: 0.6.0+cpu
torch_version: 1.5.0
- torch: 1.5.0+cu101
torchvision: 0.6.0+cu101
torch_version: 1.5.0
cuda_version: 101
- torch: 1.6.0+cu101
torchvision: 0.7.0+cu101
torch_version: 1.6.0
cuda_version: 101
- torch: 1.7.0+cu101
torchvision: 0.8.1+cu101
torch_version: 1.7.0
cuda_version: 101
- torch: 1.8.0+cu101
torchvision: 0.9.0+cu101
- torch: 1.8.0+cu101
torchvision: 0.9.0+cu101

torch_version: 1.8.0
cuda_version: 101
steps:
- uses: actions/checkout@v2
- name: Set up Python ${{ matrix.python-version }}
Expand All @@ -76,9 +83,18 @@ jobs:
run: pip install Pillow==6.2.2
- name: Install PyTorch
run: pip install torch==${{matrix.torch}} torchvision==${{matrix.torchvision}} -f https://download.pytorch.org/whl/torch_stable.html
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We need to install different device pytorch version.

- name: Install MMCV_CPU
if: ${{matrix.torch == '1.3.0+cpu' || matrix.torch == '1.5.0+cpu'}}
run: |
pip install --use-deprecated=legacy-resolver mmcv-full -f https://download.openmmlab.com/mmcv/dist/cpu/torch${{matrix.torch_version}}/index.html
python -c 'import mmcv; print(mmcv.__version__)'
- name: Install MMCV_GPU
if: ${{matrix.torch == '1.5.0+cu101' || matrix.torch == '1.6.0+cu101' || matrix.torch == '1.7.0+cu101' || matrix.torch == '1.8.0+cu101'}}
run: |
pip install --use-deprecated=legacy-resolver mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu${{matrix.cuda_version}}/torch${{matrix.torch_version}}/index.html
python -c 'import mmcv; print(mmcv.__version__)'
- name: Install mmseg dependencies
run: |
pip install mmcv-full==latest+torch${{matrix.torch}} -f https://download.openmmlab.com/mmcv/dist/index.html --use-deprecated=legacy-resolver
pip install -r requirements.txt
- name: Build and install
run: rm -rf .eggs && pip install -e .
Expand Down