xmba15 / onnx_runtime_cpp

small c++ library to quickly deploy models using onnxruntime
MIT License
327 stars 49 forks source link

Question about TensorRT environment #47

Open RUiN-jiarun opened 1 year ago

RUiN-jiarun commented 1 year ago

Hi! Thanks for your great work. The onnx_runtime_gpu version works fine to me, and I'm trying to deploy a faster TensorRT version.
I've followed the instructions of onnxruntime_tensorrt.md to build a TensorRT docker, however there are errors while building onnx_runtime. I want to know which version of onnx_runtime and TensorRT docker tag did you use to test?
BTW, the TensoRT docker tag in your dockerfile is 21.07-py3, and the script install_onnx_runtime.bash seems to install an onnx_runtime version v1.14.1.
Looking forward to your reply!