安装TensorRT一定要注意安装的版本,尤其是注意与CUDA、cuDNN、ONNX等版本的对应(在新版本的TensorRT中cuDNN已经不是必须项)。下面主要介绍使用tar包安装的方法,官方文档中还有其他安装方法可以参考。
下载
在https://developer.nvidia.com/tensorrt下载tar包,点击download now:
选择要下载的TensorRT版本:
选择匹配自己系统的tar包:
安装
参照官方文档中的安装教程进行安装:
安装步骤还可以可以参照:https://www.cnblogs.com/zycheng/articles/17670625.html