我使用的docker镜像是 nvcr.io/nvidia/tensorrt 19.03-py3 6053535b43bc
首先安装 apt-get install libprotobuf-dev protobuf-compiler
,满足环境protobuf
下载onnx-tensorrt最新master分支,使用cmake命令如下:
cmake -DCUDA_INCLUDE_DIRS=/usr/local/cuda/include -DTENSORRT_ROOT=/opt/tensorrt .. -DGPU_ARCHS="61"
当然,我gpu是1070ti,所以是61,其余的自己查看
DCUDA_INCLUDE_DIRS也要添加,否则会有drive_type.h找不到
cv2
cv2需要一些Lib库才能运行,所以需要添加
apt-get install libsm6
apt-get install libxrender1
apt-get install libxext-dev