先放上一张我运行成功的截图,只要跟着我的教程一步一步按操作,下载好匹配的软件是一定可以成功的!
我相信想要在C++平台使用tensorrt加速的朋友们也是有很强的计算机基础的,那么简单的部分我们就跳过,重点是和大家介绍模型转换的部分以及环境的搭建。
一. 环境
我的cudn版本
我的cudnn版本是8.1.1 那么我下载的对应的tensorrt版本是7.2.2.3 ,当然并不是要严格安装我的版本来,但是如果和我的版本一样那么成功的概率会很大。具体tensorrt支持的cuda和cudnn可以在官网上查看。里面有支持的各种版本,网上安装tensorrt的教程很多这里就不多介绍了。
tensorrt下载官网