You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

6 lines
197 B

__version__ = '0.18.1'
git_version = '126fc22ce33e6c2426edcf9ed540810c178fe9ce'
from torchvision.extension import _check_cuda_version
if _check_cuda_version() > 0:
cuda = _check_cuda_version()