6 lines
200 B
Python
6 lines
200 B
Python
|
__version__ = '0.9.1+cpu'
|
||
|
|
git_version = '8fb5838ca916fd4ace080dae0357e7c307037bef'
|
||
|
|
from torchvision.extension import _check_cuda_version
|
||
|
|
if _check_cuda_version() > 0:
|
||
|
|
cuda = _check_cuda_version()
|