前言
当前图片识别很慢,单张图像要60ms,考虑基于python调用c++的方法集成tensorrt,进而做推理加速。现在是第一步,本地安装tensorrt
1、安装cuda、cudnn
去英伟达官网下载cuda、cudnn、tensorrt,要注意版本。看下面链接。cudnn、tensorrt版本都是跟cuda相关。
https://developer.nvidia.com/cuda-downloads
https://developer.nvidia.com/rdp/cudnn-archive
https://developer.nvidia.com/tensorrt/download
2、cudnn安装
如下是cudnn、tensorrt 安装测试连接,已经conda 内安装tensorrt 相关whl并测试