From e971cfb71ac88ff1be6abd5bb8b4f04fdf110a88 Mon Sep 17 00:00:00 2001
From: RioShiina47 <61770365+RioShiina47@users.noreply.github.com>
Date: Tue, 16 Sep 2025 20:44:51 +0800
Subject: [PATCH] fix cudart

---
 trt_utilities.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/trt_utilities.py b/trt_utilities.py
index fcc1f01..eaa680b 100644
--- a/trt_utilities.py
+++ b/trt_utilities.py
@@ -16,7 +16,7 @@ import tensorrt as trt
 from logging import error, warning
 from tqdm import tqdm
 import copy
-from cuda import cudart
+import cuda.bindings.runtime as cudart
 
 TRT_LOGGER = trt.Logger(trt.Logger.ERROR)
 G_LOGGER.module_severity = G_LOGGER.ERROR
@@ -274,3 +274,4 @@ class Engine:
                 raise ValueError("ERROR: inference failed.")
 
         return self.tensors
+