6 lines
201 B
Python
6 lines
201 B
Python
|
__version__ = '0.22.1+cpu'
|
||
|
|
git_version = '59a3e1f9f78cfe44cb989877cc6f4ea77c8a75ca'
|
||
|
|
from torchvision.extension import _check_cuda_version
|
||
|
|
if _check_cuda_version() > 0:
|
||
|
|
cuda = _check_cuda_version()
|