[Feature]: Add cuda11.1 CI (#104)

* add cuda11.1 config

* add onnx

* update

* remove onnx

* lock mmcls version

.github/workflows/build.yml

@@ -52,7 +52,7 @@ jobs:
           python -c 'import mmcv; print(mmcv.__version__)'
       - name: Install unittest dependencies
         run: |
-          pip install -r requirements/tests.txt -r requirements/optional.txt
+          pip install -r requirements.txt
       - name: Build and install
         run: rm -rf .eggs && pip install -e .
       - name: Run unittests and generate coverage report
@@ -60,3 +60,60 @@ jobs:
           coverage run --branch --source mmdeploy -m pytest -rsE tests/
           coverage xml
           coverage report -m
+
+  build_cuda111:
+    runs-on: ubuntu-18.04
+    container:
+      image: pytorch/pytorch:1.8.0-cuda11.1-cudnn8-devel
+
+    strategy:
+      matrix:
+        python-version: [3.7]
+        torch:
+          [
+            1.8.0+cu111,
+          ]
+        include:
+          - torch: 1.8.0+cu111
+            torch_version: torch1.8.0
+            torchvision: 0.9.0+cu111
+            mmcv_link: "torch1.8.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: Install system dependencies
+        run: |
+          apt-get update && apt-get install -y ffmpeg libsm6 libxext6 git ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6 python${{matrix.python-version}}-dev
+          apt-get clean
+          rm -rf /var/lib/apt/lists/*
+      - name: Install PyTorch
+        run: python -m pip install torch==${{matrix.torch}} torchvision==${{matrix.torchvision}} -f https://download.pytorch.org/whl/torch_stable.html
+      - name: Install dependencies
+        run: |
+          python -V
+          python -m pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/${{matrix.mmcv_link}}/index.html
+          python -m pip install pycocotools
+          python -m pip install -r requirements.txt
+          python -c 'import mmcv; print(mmcv.__version__)'
+      - 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 mmdeploy -m pytest -rsE 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

requirements/optional.txt

@@ -1,8 +1,7 @@
-mmcls
+mmcls>=0.15.0
 mmdet
 mmedit
 mmocr>=0.3.0
-mmocr
 mmsegmentation
 ncnn
 onnxruntime>=1.8.0