Ubuntu 安装Tensorrt

  • 【丁丁历险记】全篇索引​​​​​​

  • 本节教大家配置在Ubuntu中安装并测试Tensorrt。由于网上各种水文,我就不讲官方文档里的那些介绍了。直接上干货。
  • 配置Tensorrt之前需要保证主机中已经安装好了NV的显卡驱动、Cuda、Cudnn(特别注意:这四个东西是必须对应版本的,这是一套东西)请在官网自行装好,个人建议安装.run文件的cuda版本,手动安装cudnn。主要是知道文件去了哪心理踏实
  • 【官方cuda链接】;【官方cudnn链接】;【nvidia显卡驱动链接】;【官方tensorrt链接】

一、下载特定版本Tensorrt

  • 每个Tensorrt的版本都有支持不同的cuda、cudnn、和显卡驱动型号

  • 根据文件名可以分辨出支持的cuda版本和系统版本进行安装。

Ubuntu 安装Tensorrt_第1张图片

                                 (个人建议还是下tar包的文件自己解压安装,如果你是deb包的就不用往下看了) 

一、解压并安装

  • 将下载好的Tensorrt--.tar 在主机内进行解压

    tar -xvf Tensorrt--.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__

    显示对应版本

 

 

你可能感兴趣的:(深度学习,tensorrt,trt)