[Fix] Fix support for devices other than Cuda in FreeAnchor3DHead #967
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
| name: pr_stage_test | |
| on: | |
| pull_request: | |
| paths-ignore: | |
| - 'README.md' | |
| - 'README_zh-CN.md' | |
| - 'docs/**' | |
| - 'demo/**' | |
| - '.dev_scripts/**' | |
| - '.circleci/**' | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.ref }} | |
| cancel-in-progress: true | |
| jobs: | |
| build_cpu: | |
| runs-on: ubuntu-22.04 | |
| strategy: | |
| matrix: | |
| python-version: [3.7] | |
| include: | |
| - torch: 1.8.1 | |
| torchvision: 0.9.1 | |
| steps: | |
| - uses: actions/checkout@v3 | |
| - name: Set up Python ${{ matrix.python-version }} | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| - name: Upgrade pip | |
| run: python -m pip install pip --upgrade && pip install wheel | |
| - name: Install PyTorch | |
| run: pip install torch==${{matrix.torch}}+cpu torchvision==${{matrix.torchvision}}+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html | |
| - name: Install MMEngine | |
| run: pip install git+https://github.com/open-mmlab/mmengine.git@main | |
| - name: Install MMCV | |
| run: | | |
| pip install -U openmim | |
| mim install 'mmcv >= 2.0.0rc4' | |
| - name: Install MMDet | |
| run: mim install 'mmdet>=3.0.0' | |
| - name: Install other dependencies | |
| run: pip install -r requirements/tests.txt | |
| - name: Build and install | |
| run: rm -rf .eggs && pip install -e . | |
| - name: Run unittests and generate coverage report | |
| run: | | |
| coverage run --branch --source mmdet3d -m pytest tests/ | |
| coverage xml | |
| coverage report -m | |
| # Upload coverage report for python3.7 && pytorch1.8.1 cpu | |
| - name: Upload coverage to Codecov | |
| uses: codecov/[email protected] | |
| with: | |
| file: ./coverage.xml | |
| flags: unittests | |
| env_vars: OS,PYTHON | |
| name: codecov-umbrella | |
| fail_ci_if_error: false | |
| build_cu102: | |
| runs-on: ubuntu-22.04 | |
| container: | |
| image: pytorch/pytorch:1.8.1-cuda10.2-cudnn7-devel | |
| strategy: | |
| matrix: | |
| python-version: [3.7] | |
| steps: | |
| - uses: actions/checkout@v3 | |
| - name: Set up Python ${{ matrix.python-version }} | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| - name: Upgrade pip | |
| run: pip install pip --upgrade && pip install wheel | |
| - name: Fetch GPG keys | |
| run: | | |
| apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub | |
| apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub | |
| - 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 | |
| - name: Install mmdet3d dependencies | |
| run: | | |
| pip install git+https://github.com/open-mmlab/mmengine.git@main | |
| pip install -U openmim | |
| mim install 'mmcv >= 2.0.0rc4' | |
| mim install 'mmdet>=3.0.0' | |
| pip install -r requirements/tests.txt | |
| - name: Build and install | |
| run: pip install -e . | |
| - name: Run unittests and generate coverage report | |
| run: | | |
| coverage run --branch --source mmdet3d -m pytest tests/ | |
| coverage xml | |
| coverage report -m | |
| build_windows: | |
| runs-on: windows-2022 | |
| strategy: | |
| matrix: | |
| python-version: [3.7] | |
| platform: [cpu, cu111] | |
| torch: [1.8.1] | |
| torchvision: [0.9.1] | |
| steps: | |
| - uses: actions/checkout@v3 | |
| - name: Set up Python ${{ matrix.python-version }} | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| - name: Upgrade pip | |
| run: python -m pip install pip --upgrade && pip install wheel | |
| - name: Install lmdb | |
| run: pip install lmdb | |
| - name: Install PyTorch | |
| run: pip install torch==${{matrix.torch}}+${{matrix.platform}} torchvision==${{matrix.torchvision}}+${{matrix.platform}} -f https://download.pytorch.org/whl/${{matrix.platform}}/torch_stable.html | |
| - name: Install mmdet3d dependencies | |
| run: | | |
| pip install git+https://github.com/open-mmlab/mmengine.git@main | |
| pip install -U openmim | |
| mim install 'mmcv >= 2.0.0rc4' | |
| mim install 'mmdet>=3.0.0' | |
| pip install -r requirements/tests.txt | |
| - name: Build and install | |
| run: pip install -e . | |
| - name: Run unittests and generate coverage report | |
| run: pytest tests/ |