Description
When I generated the engine for the wts model of yolov5-v6.0 in TensorRT8.5.2, an error occurred.
A clear and concise description of the bug or issue.
Environment
TensorRT Version: 8.5.2
GPU Type: Jetson Xavier
Nvidia Driver Version: Jetpack5.1.1
CUDA Version: 11.4
CUDNN Version: 8.6
Operating System + Version:arrch64+ubuntu20.4
Python Version (if applicable): 3.8
PyTorch Version (if applicable): 2.0.0
When I run sudo ./yolov5 -s …/best.wts …/best.engine s the following error appears
yolov5: /opt/tensorrtx-yolov5-v6.0/yolov5/yolov5.cpp:63: nvinfer1::ICudaEngine* build_engine(unsigned int, nvinfer1::IBuilder*, nvinfer1::IBuilderConfig*, nvinfer1::DataType, float&, float&, std::string&): Assertion `upsample11’ failed.
已放弃