We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 59ab66b commit 7f869c7Copy full SHA for 7f869c7
test/test_transforms_v2.py
@@ -52,11 +52,11 @@
52
from torchvision.transforms.v2 import functional as F
53
from torchvision.transforms.v2._utils import check_type, is_pure_tensor
54
from torchvision.transforms.v2.functional._geometry import _get_perspective_coeffs, _parallelogram_to_bounding_boxes
55
+from torchvision.transforms.v2.functional._type_conversion import _import_cvcuda_modules
56
from torchvision.transforms.v2.functional._utils import _get_kernel, _register_kernel_internal
57
58
try:
- import cvcuda # type: ignore[import-not-found]
59
-
+ _import_cvcuda_modules()
60
CVCUDA_AVAILABLE = True
61
except ImportError:
62
CVCUDA_AVAILABLE = False
0 commit comments