Function is_cuda_available

install_utils.py:45–56

Check if CUDA is available on the system by attempting to get the CUDA version. Returns: True if CUDA is available and supported, False otherwise.

()

def is_cuda_available() -> bool:
    """
    Check if CUDA is available on the system by attempting to get the CUDA version.

    Returns:
        True if CUDA is available and supported, False otherwise.
    """
    try:
        _get_cuda_version()
        return True
    except Exception:
        return False


@functools.lru_cache(maxsize=1)