mirror of https://github.com/open-mmlab/mmcv.git
14 lines
344 B
Python
14 lines
344 B
Python
|
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
|