Bug fix: decode error_string if necessary. (#859)

Signed-off-by: SimengLiu-nv <[email protected]>
onnx · Jul 25, 2022 · c3cfcbc · c3cfcbc
1 parent 1da7332
commit c3cfcbc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/onnx_tensorrt/backend.py b/onnx_tensorrt/backend.py
@@ -18,6 +18,8 @@ def cudaSetDevice(device_idx):
     ret = libcudart.cudaSetDevice(device_idx)
     if ret != 0:
         error_string = libcudart.cudaGetErrorString(ret)
+        if isinstance(error_string, bytes):
+            error_string = error_string.decode("utf-8")
         raise RuntimeError("cudaSetDevice: " + error_string)
 
 def count_trailing_ones(vals):