diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 28056d3f05..c16cb73955 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -32,7 +32,7 @@ jobs: strategy: matrix: python-version: [3.7] - torch: [1.3.0, 1.5.0, 1.6.0, 1.7.0] + torch: [1.3.0, 1.5.0, 1.6.0, 1.7.0, 1.8.0, 1.9.0] include: - torch: 1.3.0 torchvision: 0.4.1 @@ -42,6 +42,10 @@ jobs: torchvision: 0.7.0 - torch: 1.7.0 torchvision: 0.8.1 + - torch: 1.8.0 + torchvision: 0.9.0 + - torch: 1.9.0 + torchvision: 0.10.0 steps: - uses: actions/checkout@v2 - name: Set up Python ${{ matrix.python-version }} @@ -67,7 +71,7 @@ jobs: coverage xml coverage report -m - build_cuda: + build_cuda101: runs-on: ubuntu-latest env: @@ -77,7 +81,7 @@ jobs: strategy: matrix: python-version: [3.7] - torch: [1.3.0, 1.5.0, 1.6.0, 1.7.0] + torch: [1.3.0, 1.5.0, 1.6.0, 1.7.0, 1.8.0] include: - torch: 1.3.0 torchvision: 0.4.1 @@ -93,6 +97,8 @@ jobs: - torch: 1.7.0 torchvision: 0.8.1 python-version: 3.8 + - torch: 1.8.0 + torchvision: 0.9.0 steps: - uses: actions/checkout@v2 @@ -148,3 +154,68 @@ jobs: env_vars: OS,PYTHON name: codecov-umbrella fail_ci_if_error: false + + + build_cuda102: + runs-on: ubuntu-latest + + env: + CUDA: 10.2.89-1 + CUDA_SHORT: 10.2 + UBUNTU_VERSION: ubuntu1804 + strategy: + matrix: + python-version: [3.7] + torch: [1.9.0] + include: + - torch: 1.9.0 + torchvision: 0.10.0 + + steps: + - uses: actions/checkout@v2 + - name: Set up Python ${{ matrix.python-version }} + uses: actions/setup-python@v2 + with: + python-version: ${{ matrix.python-version }} + - name: Upgrade pip + run: pip install pip --upgrade + - name: Install CUDA + run: | + export INSTALLER=cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb + wget http://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/${INSTALLER} + sudo dpkg -i ${INSTALLER} + wget https://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/7fa2af80.pub + sudo apt-key add 7fa2af80.pub + sudo apt update -qq + sudo apt install -y cuda-${CUDA_SHORT/./-} cuda-cufft-dev-${CUDA_SHORT/./-} + sudo apt clean + export CUDA_HOME=/usr/local/cuda-${CUDA_SHORT} + export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${CUDA_HOME}/include:${LD_LIBRARY_PATH} + export PATH=${CUDA_HOME}/bin:${PATH} + - name: Install Pillow + run: pip install Pillow==6.2.2 + if: ${{matrix.torchvision < 0.5}} + - name: Install PyTorch + run: pip install torch==${{matrix.torch}}+cu102 torchvision==${{matrix.torchvision}}+cu102 -f https://download.pytorch.org/whl/torch_stable.html + - name: Install MMCV + run: pip install mmcv-full==1.3.0 -f https://download.openmmlab.com/mmcv/dist/cu101/torch${{matrix.torch}}/index.html + - name: Install other dependencies + run: pip install -r requirements.txt + - name: Build and install + run: | + rm -rf .eggs + python setup.py check -m -s + TORCH_CUDA_ARCH_LIST=7.0 pip install . + - name: Run unittests and generate coverage report + run: | + coverage run --branch --source mmedit -m pytest tests/ + coverage xml + coverage report -m + - name: Upload coverage to Codecov + uses: codecov/codecov-action@v1.0.10 + with: + file: ./coverage.xml + flags: unittests + env_vars: OS,PYTHON + name: codecov-umbrella + fail_ci_if_error: false