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
6 lines
197 B
5 months ago
|
__version__ = '0.18.1'
|
||
|
|
git_version = '126fc22ce33e6c2426edcf9ed540810c178fe9ce'
|
||
|
|
from torchvision.extension import _check_cuda_version
|
||
|
|
if _check_cuda_version() > 0:
|
||
|
|
cuda = _check_cuda_version()
|