@@ -295,6 +295,52 @@ jobs:
295
295
coverage xml
296
296
if : ${{matrix.python-version == '3.10'}}
297
297
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
298
344
299
345
build_windows_without_ops :
300
346
runs-on : windows-latest
0 commit comments