delete _is_cvcuda_available function

Author: AntoineSimoulin

test/test_transforms_v2.py

@@ -44,7 +44,7 @@
 from torch.testing import assert_close
 from torch.utils._pytree import tree_flatten, tree_map
 from torch.utils.data import DataLoader, default_collate
-from torchvision import _is_cvcuda_available, tv_tensors
+from torchvision import tv_tensors
 from torchvision.ops.boxes import box_iou
 
 from torchvision.transforms._functional_tensor import _max_value as get_max_value
@@ -54,8 +54,12 @@
 from torchvision.transforms.v2.functional._geometry import _get_perspective_coeffs, _parallelogram_to_bounding_boxes
 from torchvision.transforms.v2.functional._utils import _get_kernel, _register_kernel_internal
 
+try:
+    import cvcuda  # type: ignore[import-not-found]
 
-CVCUDA_AVAILABLE = _is_cvcuda_available()
+    CVCUDA_AVAILABLE = True
+except ImportError:
+    CVCUDA_AVAILABLE = False
 CUDA_AVAILABLE = torch.cuda.is_available()
 
 

torchvision/__init__.py

@@ -99,15 +99,6 @@ def _is_tracing():
     return torch._C._get_tracing_state()
 
 
-def _is_cvcuda_available() -> bool:
-    try:
-        import cvcuda  # type: ignore[import-not-found]
-
-        return True
-    except ImportError:
-        return False
-
-
 def disable_beta_transforms_warning():
     # Noop, only exists to avoid breaking existing code.
     # See https://github.com/pytorch/vision/issues/7896