下载TensorRT
在官网下载TensorRT。我的cuda版本是12.2,因此选择的TensorRT版本是10.7。但需要注意的是,文档中的示例使用的是TensorRT 10.6.0版本,安装步骤是通用的,只需根据具体版本进行调整。
由于下载.deb文件进行安装可能会遇到问题,因此推荐下载.gz.tar压缩包。
安装TensorRT
-
解压
tar -zvxf TensorRT-10.6.0.26.Linux.x86_64-gnu.cuda-12.6.tar.gz
(注意:此处版本号需与下载的TensorRT版本一致)
-
添加环境变量
编辑
~/.bashrc
文件,添加以下行:export LD_LIBRARY_PATH=<TensorRT-${version}/lib>:$LD_LIBRARY_PATH
(将
<TensorRT-${version}/lib>
替换为实际的TensorRT库路径)然后,运行
source ~/.bashrc
使更改生效。
测试TensorRT
-
编译示例代码
进入TensorRT安装目录下的samples目录,例如: