Use CUDA-Python to get CUDA runtime version

Author: jakirkham

python/cugraph/cugraph/tests/data_store/test_gnn_feat_storage_wholegraph.py

@@ -15,7 +15,7 @@
 import numpy as np
 import os
 
-import numba.cuda
+from cuda.bindings import runtime as cuda_runtime
 
 from cugraph.gnn import FeatureStore
 
@@ -28,8 +28,10 @@
 
 
 def get_cudart_version():
-    major, minor = numba.cuda.runtime.get_version()
-    return major * 1000 + minor * 10
+    status, version = cuda_runtime.getLocalRuntimeVersion()
+    if status != cuda_runtime.cudaError_t.cudaSuccess:
+        raise RuntimeError(f"CUDA Error: {status}")
+    return version
 
 
 pytestmark = [