tensorrtx
tensorrtx copied to clipboard
the command : ./yolov5_det -s yolov5s.wts yolov5s.engine s, error!!!
Env
- GPU: RTX3090
- OS: Ubuntu 20.04
- Cuda version: 11.4
- TensorRT version: 8.5.2.2
- torch: 1.12.0
About this repo
- branch: tensorrtx(yolov5-v7.0)
- model: yolov5s, v7.0, official weight
Your problem
-
my command: ./yolov5_det -s yolov5s.wts yolov5s.engine s
-
output: [E] [TRT] 9: [graphShapeAnalyzer.cpp::addVolumeCheck::739] Error Code 9: Internal Error (prob: dimension 0 never exceeds -5) [05/24/2024-19:01:38] [E] [TRT] 4: [network.cpp::validate::2980] Error Code 4: Internal Error (Could not compute dimensions for (Unnamed Layer* 0) [Convolution]_output, because the network is not valid.)
-
expect output: yolov5s.engine