From dda642f0db7bae72b6df877633e2638fe6ba8de1 Mon Sep 17 00:00:00 2001
From: YunLiu <55491388+KumoLiu@users.noreply.github.com>
Date: Wed, 28 Aug 2024 17:40:23 +0800
Subject: [PATCH] fix #8050

Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com>
---
 monai/networks/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/monai/networks/utils.py b/monai/networks/utils.py
index f301c2dd5c..bd65ffa33e 100644
--- a/monai/networks/utils.py
+++ b/monai/networks/utils.py
@@ -851,7 +851,7 @@ def _onnx_trt_compile(
     # wrap the serialized TensorRT engine back to a TorchScript module.
     trt_model = torch_tensorrt.ts.embed_engine_in_new_module(
         f.getvalue(),
-        device=torch.device(f"cuda:{device}"),
+        device=torch_tensorrt.Device(f"cuda:{device}"),
         input_binding_names=input_names,
         output_binding_names=output_names,
     )