1
0
mirror of https://github.com/open-mmlab/mmcv.git synced 2025-06-03 21:54:52 +08:00

15 lines
384 B
Python
Raw Normal View History

# Copyright (c) OpenMMLab. All rights reserved.
import torch
class TestInfo:
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