- 【丁丁历险记】全篇索引
- 本节教大家配置在Ubuntu中安装并测试Tensorrt。由于网上各种水文,我就不讲官方文档里的那些介绍了。直接上干货。
- 配置Tensorrt之前需要保证主机中已经安装好了NV的显卡驱动、Cuda、Cudnn(特别注意:这四个东西是必须对应版本的,这是一套东西)请在官网自行装好,个人建议安装.run文件的cuda版本,手动安装cudnn。主要是知道文件去了哪心理踏实。
- 【官方cuda链接】;【官方cudnn链接】;【nvidia显卡驱动链接】;【官方tensorrt链接】
一、下载特定版本Tensorrt
-
每个Tensorrt的版本都有支持不同的cuda、cudnn、和显卡驱动型号
- 根据文件名可以分辨出支持的cuda版本和系统版本进行安装。
(个人建议还是下tar包的文件自己解压安装,如果你是deb包的就不用往下看了)
一、解压并安装
-
将下载好的Tensorrt-<version>-.tar 在主机内进行解压
tar -xvf Tensorrt-<version>-.tar
之后将解压后文件夹中的lib路径添加到.bashrc中
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/youName/TensorRT-version/lib
现在source一下.bashrc就已经可以使用了
-
安装python的支持
在你的Tensorrt-version/python文件夹下有许多的whl文件,直接使用对应版本的pip进行安装即可添加对python的文件支持。 -
测试安装
import tensorrt as trt trt.__version__
显示对应版本