tvm_tirx.support.nvcc.get_cuda_version

tvm_tirx.support.nvcc.get_cuda_version(cuda_path=None)

Utility function to get CUDA version

Parameters:

cuda_path (Optional[str]) – Path to CUDA root. If None is passed, will use find_cuda_path() as default.

Returns:

version (float) – The CUDA version