ttanzhiqiang / onnx_tensorrt_project

Support Yolov5(4.0)/Yolov5(5.0)/YoloR/YoloX/Yolov4/Yolov3/CenterNet/CenterFace/RetinaFace/Classify/Unet. use darknet/libtorch/pytorch/mxnet to onnx to tensorrt
210 stars 43 forks source link

how to run? #5

Closed xinsuinizhuan closed 3 years ago

xinsuinizhuan commented 3 years ago

my windows eviroment is: cuda 11.0 vs2019 how to run? when i run, it tip me: can not find the nvrtc64_111_0.dll

which third_model should i to recompile them?

ttanzhiqiang commented 3 years ago

you should use cuda11.1

xinsuinizhuan commented 3 years ago

you should use cuda11.1

If i use the cuda11.0, which 3rdparty should i recompile?

ttanzhiqiang commented 3 years ago

you should change tensorrt for cuda11.0

xinsuinizhuan commented 3 years ago

OK.thank you. It works, now .but where i run the tiny_tensorrt_dyn_onnx, it have no: model\yolov5\yolov5-v5_weight\yolov5m.onnx

ttanzhiqiang commented 3 years ago

look https://share.weiyun.com/CJCwngAM. and tiny_tensorrt_dyn_onnx is dynamic shape,you should use tiny_tensorrt_onnx.sln

xinsuinizhuan commented 3 years ago

tiny_tensorrt_onnx.sln

yes, i run it, but now the have no model of the yolov5, could you share them? look, have no model of yolov5 图片