mmcv/tests/test_ops/test_info.py

# Copyright (c) OpenMMLab. All rights reserved.
import torch


class TestInfo(object):

    def test_info(self):
        if not torch.cuda.is_available():
            return
        from mmcv.ops import get_compiler_version, get_compiling_cuda_version
        cv = get_compiler_version()
        ccv = get_compiling_cuda_version()
        assert cv is not None
        assert ccv is not None