You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Device Name : Jetson Xavier Nx DeveloperToolkit
Jetpack:5.1.1
The inference code is running fine but on end of the prediction it displays following error message
[09/25/2023-10:36:28] [TRT] [E] 1: [defaultAllocator.cpp::deallocate::42] Error Code 1: Cuda Runtime (invalid argument). Is it something we should be worried about ?
Loading nms.trt for TensorRT inference...
[09/25/2023-10:15:14] [TRT] [I] Loaded engine size: 147 MiB
[09/25/2023-10:15:15] [TRT] [W] Using an engine plan file across different models of devices is not recommended and is likely to affect performance or even cause errors.
[09/25/2023-10:15:16] [TRT] [I] [MemUsageChange] Init cuBLAS/cuBLASLt: CPU +261, GPU +247, now: CPU 750, GPU 4081 (MiB)
[09/25/2023-10:15:17] [TRT] [I] [MemUsageChange] Init cuDNN: CPU +82, GPU +86, now: CPU 832, GPU 4167 (MiB)
[09/25/2023-10:15:17] [TRT] [I] [MemUsageChange] TensorRT-managed allocation in engine deserialization: CPU +0, GPU +150, now: CPU 0, GPU 150 (MiB)
[09/25/2023-10:15:20] [TRT] [I] [MemUsageChange] Init cuBLAS/cuBLASLt: CPU +0, GPU +7, now: CPU 1307, GPU 4716 (MiB)
[09/25/2023-10:15:20] [TRT] [I] [MemUsageChange] Init cuDNN: CPU +1, GPU +10, now: CPU 1308, GPU 4726 (MiB)
[09/25/2023-10:15:20] [TRT] [I] [MemUsageChange] TensorRT-managed allocation in IExecutionContext creation: CPU +0, GPU +170, now: CPU 0, GPU 320 (MiB)
123
[09/25/2023-10:15:22] [TRT] [E] 1: [defaultAllocator.cpp::deallocate::42] Error Code 1: Cuda Runtime (invalid argument)
Segmentation fault (core dumped)
The code is as follows:
import torch
from yolort.yolort.runtime import PredictorTRT
# Load the serialized TensorRT engine
engine_path = "nms.trt"
device = torch.device("cuda")
y_runtime = PredictorTRT(engine_path, device=device)
# Perform inference on an image file
predictions = y_runtime.predict("dummy_frame.jpg")
print("123")
The text was updated successfully, but these errors were encountered:
Device Name : Jetson Xavier Nx DeveloperToolkit
Jetpack:5.1.1
The inference code is running fine but on end of the prediction it displays following error message
[09/25/2023-10:36:28] [TRT] [E] 1: [defaultAllocator.cpp::deallocate::42] Error Code 1: Cuda Runtime (invalid argument). Is it something we should be worried about ?
The text was updated successfully, but these errors were encountered: