mirror of https://github.com/open-mmlab/mmcv.git
Add torch1.13 checking in CI (#2405)
* Add torch1.13 checking in CI * swith to python3.10 * fix * cancel commentspull/2420/head
parent
12442667ff
commit
eda3c57ea1
|
@ -295,6 +295,52 @@ jobs:
|
|||
coverage xml
|
||||
if: ${{matrix.python-version == '3.10'}}
|
||||
|
||||
build_cu116:
|
||||
runs-on: ubuntu-18.04
|
||||
container:
|
||||
image: pytorch/pytorch:1.13.0-cuda11.6-cudnn8-devel
|
||||
env:
|
||||
FORCE_CUDA: 1
|
||||
MMCV_CUDA_ARGS: -gencode=arch=compute_61,code=sm_61
|
||||
strategy:
|
||||
matrix:
|
||||
python-version: ['3.10']
|
||||
torch: [1.13.0+cu116]
|
||||
include:
|
||||
- torch: 1.13.0+cu116
|
||||
torchvision: 0.14.0+cu116
|
||||
steps:
|
||||
- uses: actions/checkout@v2
|
||||
- name: Set up Python ${{ matrix.python-version }}
|
||||
uses: actions/setup-python@v2
|
||||
with:
|
||||
python-version: ${{ matrix.python-version }}
|
||||
- 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: Add PPA
|
||||
run: |
|
||||
apt-get update && apt-get install -y software-properties-common
|
||||
add-apt-repository -y ppa:deadsnakes/ppa
|
||||
- name: Install python-dev
|
||||
run: apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y python${{matrix.python-version}}-dev
|
||||
- name: python -m Install PyTorch
|
||||
run: python -m pip install torch==${{matrix.torch}} torchvision==${{matrix.torchvision}} -f https://download.pytorch.org/whl/torch_stable.html
|
||||
- name: Install system dependencies
|
||||
run: apt-get update && apt-get install -y ffmpeg libturbojpeg ninja-build
|
||||
# pstuil is an optional package to detect the number of CPU for compiling mmcv
|
||||
- name: Install psutil
|
||||
run: python -m pip install psutil
|
||||
- name: Build and install
|
||||
run: rm -rf .eggs && python -m pip install -e .
|
||||
- name: Validate the installation
|
||||
run: python -c "import mmcv"
|
||||
- name: Run unittests and generate coverage report
|
||||
run: |
|
||||
python -m pip install -r requirements/test.txt
|
||||
coverage run --branch --source=mmcv -m pytest tests/ --ignore=tests/test_ops/test_onnx.py --ignore=tests/test_ops/test_tensorrt.py --ignore=tests/test_ops/test_tensorrt_preprocess.py
|
||||
coverage xml
|
||||
|
||||
build_windows_without_ops:
|
||||
runs-on: windows-latest
|
||||
|
|
Loading…
Reference in New Issue