-
Notifications
You must be signed in to change notification settings - Fork 2.6k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge remote-tracking branch 'upstream/master' into bump_0.29.2
- Loading branch information
Showing
5 changed files
with
262 additions
and
152 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,157 +1,34 @@ | ||
version: 2.1 | ||
|
||
jobs: | ||
lint: | ||
docker: | ||
- image: cimg/python:3.7.4 | ||
steps: | ||
- checkout | ||
- run: | ||
name: Install dependencies | ||
command: | | ||
sudo apt-add-repository ppa:brightbox/ruby-ng -y | ||
sudo apt-get update | ||
sudo apt-get install -y ruby2.7 | ||
- run: | ||
name: Install pre-commit hook | ||
command: | | ||
pip install pre-commit | ||
pre-commit install | ||
- run: | ||
name: Linting | ||
command: pre-commit run --all-files | ||
- run: | ||
name: Check docstring coverage | ||
command: | | ||
pip install interrogate | ||
interrogate -v --ignore-init-method --ignore-module --ignore-nested-functions --ignore-regex "__repr__" --fail-under 50 mmseg | ||
# this allows you to use CircleCI's dynamic configuration feature | ||
setup: true | ||
|
||
build_cpu: | ||
parameters: | ||
# The python version must match available image tags in | ||
# https://circleci.com/developer/images/image/cimg/python | ||
python: | ||
type: string | ||
default: "3.7.4" | ||
torch: | ||
type: string | ||
torchvision: | ||
type: string | ||
docker: | ||
- image: cimg/python:<< parameters.python >> | ||
resource_class: large | ||
steps: | ||
- checkout | ||
- run: | ||
name: Install Libraries | ||
command: | | ||
sudo apt-get update | ||
sudo apt-get install -y ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6 libgl1-mesa-glx libjpeg-dev zlib1g-dev libtinfo-dev libncurses5 | ||
- run: | ||
name: Configure Python & pip | ||
command: | | ||
python -m pip install --upgrade pip | ||
python -m pip install wheel | ||
- run: | ||
name: Install PyTorch | ||
command: | | ||
python -V | ||
python -m pip install torch==<< parameters.torch >>+cpu torchvision==<< parameters.torchvision >>+cpu -f https://download.pytorch.org/whl/torch_stable.html | ||
- run: | ||
name: Install mmseg dependencies | ||
command: | | ||
python -m pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cpu/torch<< parameters.torch >>/index.html | ||
python -m pip install -r requirements.txt | ||
- run: | ||
name: Build and install | ||
command: | | ||
python -m pip install -e . | ||
- run: | ||
name: Run unittests but skip timm unittests | ||
command: | | ||
python -m coverage run --branch --source mmseg -m pytest tests/ --ignore tests/test_models/test_backbones/test_timm_backbone.py | ||
python -m coverage xml | ||
python -m coverage report -m | ||
build_cu101: | ||
machine: | ||
image: ubuntu-1604-cuda-10.1:201909-23 | ||
resource_class: gpu.nvidia.small | ||
steps: | ||
- checkout | ||
- run: | ||
name: Install Libraries | ||
command: | | ||
sudo apt-get update | ||
sudo apt-get install -y git ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6 libgl1-mesa-glx | ||
- run: | ||
name: Configure Python & pip | ||
command: | | ||
pyenv global 3.7.0 | ||
python -m pip install --upgrade pip | ||
python -m pip install wheel | ||
- run: | ||
name: Install PyTorch | ||
command: | | ||
python -V | ||
python -m pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html | ||
- run: | ||
name: Install mmseg dependencies | ||
# python -m pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch${{matrix.torch_version}}/index.html | ||
command: | | ||
python -m pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html | ||
python -m pip install -r requirements.txt | ||
- run: | ||
name: Build and install | ||
command: | | ||
python setup.py check -m -s | ||
TORCH_CUDA_ARCH_LIST=7.0 python -m pip install -e . | ||
- run: | ||
name: Run unittests but skip timm unittests | ||
command: | | ||
python -m pytest tests/ --ignore tests/test_models/test_backbones/test_timm_backbone.py | ||
# the path-filtering orb is required to continue a pipeline based on | ||
# the path of an updated fileset | ||
orbs: | ||
path-filtering: circleci/path-filtering@0.1.2 | ||
|
||
workflows: | ||
unit_tests: | ||
# the always-run workflow is always triggered, regardless of the pipeline parameters. | ||
always-run: | ||
jobs: | ||
- lint | ||
- build_cpu: | ||
name: build_cpu_th1.6 | ||
torch: 1.6.0 | ||
torchvision: 0.7.0 | ||
requires: | ||
- lint | ||
- build_cpu: | ||
name: build_cpu_th1.7 | ||
torch: 1.7.0 | ||
torchvision: 0.8.1 | ||
requires: | ||
- lint | ||
- build_cpu: | ||
name: build_cpu_th1.8_py3.9 | ||
torch: 1.8.0 | ||
torchvision: 0.9.0 | ||
python: "3.9.0" | ||
requires: | ||
- lint | ||
- build_cpu: | ||
name: build_cpu_th1.9_py3.8 | ||
torch: 1.9.0 | ||
torchvision: 0.10.0 | ||
python: "3.8.0" | ||
requires: | ||
- lint | ||
- build_cpu: | ||
name: build_cpu_th1.9_py3.9 | ||
torch: 1.9.0 | ||
torchvision: 0.10.0 | ||
python: "3.9.0" | ||
requires: | ||
- lint | ||
- build_cu101: | ||
requires: | ||
- build_cpu_th1.6 | ||
- build_cpu_th1.7 | ||
- build_cpu_th1.8_py3.9 | ||
- build_cpu_th1.9_py3.8 | ||
- build_cpu_th1.9_py3.9 | ||
# the path-filtering/filter job determines which pipeline | ||
# parameters to update. | ||
- path-filtering/filter: | ||
name: check-updated-files | ||
# 3-column, whitespace-delimited mapping. One mapping per | ||
# line: | ||
# <regex path-to-test> <parameter-to-set> <value-of-pipeline-parameter> | ||
mapping: | | ||
mmseg/.* lint_only false | ||
requirements/.* lint_only false | ||
tests/.* lint_only false | ||
tools/.* lint_only false | ||
configs/.* lint_only false | ||
.circleci/.* lint_only false | ||
base-revision: master | ||
# this is the path of the configuration we should trigger once | ||
# path filtering and pipeline parameter value updates are | ||
# complete. In this case, we are using the parent dynamic | ||
# configuration itself. | ||
config-path: .circleci/test.yml |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,11 @@ | ||
ARG PYTORCH="1.8.1" | ||
ARG CUDA="10.2" | ||
ARG CUDNN="7" | ||
|
||
FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel | ||
|
||
# To fix GPG key error when running apt-get update | ||
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub | ||
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub | ||
|
||
RUN apt-get update && apt-get install -y ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6 libgl1-mesa-glx |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,19 @@ | ||
#!/bin/bash | ||
|
||
TORCH=$1 | ||
CUDA=$2 | ||
|
||
# 10.2 -> cu102 | ||
MMCV_CUDA="cu`echo ${CUDA} | tr -d '.'`" | ||
|
||
# MMCV only provides pre-compiled packages for torch 1.x.0 | ||
# which works for any subversions of torch 1.x. | ||
# We force the torch version to be 1.x.0 to ease package searching | ||
# and avoid unnecessary rebuild during MMCV's installation. | ||
TORCH_VER_ARR=(${TORCH//./ }) | ||
TORCH_VER_ARR[2]=0 | ||
printf -v MMCV_TORCH "%s." "${TORCH_VER_ARR[@]}" | ||
MMCV_TORCH=${MMCV_TORCH%?} # Remove the last dot | ||
|
||
echo "export MMCV_CUDA=${MMCV_CUDA}" >> $BASH_ENV | ||
echo "export MMCV_TORCH=${MMCV_TORCH}" >> $BASH_ENV |
Oops, something went wrong.