Skip to content

Commit eda3c57

Browse files
authoredNov 17, 2022
Add torch1.13 checking in CI (open-mmlab#2405)
* Add torch1.13 checking in CI * swith to python3.10 * fix * cancel comments
1 parent 1244266 commit eda3c57

File tree

1 file changed

+46
-0
lines changed

1 file changed

+46
-0
lines changed
 

‎.github/workflows/build.yml

+46
Original file line numberDiff line numberDiff line change
@@ -295,6 +295,52 @@ jobs:
295295
coverage xml
296296
if: ${{matrix.python-version == '3.10'}}
297297

298+
build_cu116:
299+
runs-on: ubuntu-18.04
300+
container:
301+
image: pytorch/pytorch:1.13.0-cuda11.6-cudnn8-devel
302+
env:
303+
FORCE_CUDA: 1
304+
MMCV_CUDA_ARGS: -gencode=arch=compute_61,code=sm_61
305+
strategy:
306+
matrix:
307+
python-version: ['3.10']
308+
torch: [1.13.0+cu116]
309+
include:
310+
- torch: 1.13.0+cu116
311+
torchvision: 0.14.0+cu116
312+
steps:
313+
- uses: actions/checkout@v2
314+
- name: Set up Python ${{ matrix.python-version }}
315+
uses: actions/setup-python@v2
316+
with:
317+
python-version: ${{ matrix.python-version }}
318+
- name: Fetch GPG keys
319+
run: |
320+
apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
321+
apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
322+
- name: Add PPA
323+
run: |
324+
apt-get update && apt-get install -y software-properties-common
325+
add-apt-repository -y ppa:deadsnakes/ppa
326+
- name: Install python-dev
327+
run: apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y python${{matrix.python-version}}-dev
328+
- name: python -m Install PyTorch
329+
run: python -m pip install torch==${{matrix.torch}} torchvision==${{matrix.torchvision}} -f https://download.pytorch.org/whl/torch_stable.html
330+
- name: Install system dependencies
331+
run: apt-get update && apt-get install -y ffmpeg libturbojpeg ninja-build
332+
# pstuil is an optional package to detect the number of CPU for compiling mmcv
333+
- name: Install psutil
334+
run: python -m pip install psutil
335+
- name: Build and install
336+
run: rm -rf .eggs && python -m pip install -e .
337+
- name: Validate the installation
338+
run: python -c "import mmcv"
339+
- name: Run unittests and generate coverage report
340+
run: |
341+
python -m pip install -r requirements/test.txt
342+
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
343+
coverage xml
298344
299345
build_windows_without_ops:
300346
runs-on: windows-latest

0 commit comments

Comments
 (0)
Please sign in to comment.