Problem:
I need to deploy the pytorch model on the board, the pth was converted to onnx successfully. But when onnx was converted to trt some errors about cuda appeared.
NX has JetPack5.1.3/cuda11.4 installed. During the work,model conversion requires tensorrtversion 8.6.1, and this version of tensorrt requires cuda12.0. So I installed cuda12.0/12.2 and tensorrt8.6.1 manually, and i set the environment variables for cuda12.x and trt.
SF Environment:(JetPack5.1.3/cuda11.4/tensorrt8.5.2)
Jetson Xvaier NX ()
cuda:11.4/12.0/12.2
tensorrt:8.6.1
Onnx model is here:
pcdSeg.zip (1.8 MB)
Error: