mirror of
https://github.com/open-mmlab/mmcv.git
synced 2025-06-03 21:54:52 +08:00
15 lines
392 B
Python
15 lines
392 B
Python
# 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
|