fix: Update CUDA imports (#847)

Author: yinggeh

src/python/library/tritonclient/utils/cuda_shared_memory/__init__.py

@@ -28,7 +28,7 @@
 # Check for dependency before other import so other imports can assume
 # the module is available (drop "try ... except .."")
 try:
-    from cuda import cudart
+    import cuda.bindings.runtime as cudart
 except ModuleNotFoundError as error:
     raise RuntimeError(
         "CUDA shared memory utilities require Python package 'cuda-python'"

src/python/library/tritonclient/utils/cuda_shared_memory/_utils.py

@@ -26,8 +26,8 @@
 
 from typing import Any
 
-from cuda import cuda as cuda_driver
-from cuda import cudart
+import cuda.bindings.driver as cuda_driver
+import cuda.bindings.runtime as cudart
 
 
 def call_cuda_function(function, *argv):