Ubuntu 安装Tensorrt

1 篇文章 0 订阅
1 篇文章 0 订阅

  • 本节教大家配置在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__

    显示对应版本

 

 

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值