锐客网

  1. 首页 > it技术 > >

ubuntu20安装tensorRT

大体来说安装的方法有两种：

  • 用run安装cuda, 用wheel安装 tensorrt，好处是可以自己选驱动的版本；（大多数人选择的的方法，也是本文介绍的方法）
  • 如果想安装tensorRT的deb版本，cuda必须使用deb安装，然而cuda的deb版本安装的时候可能会覆盖你已经安装的nvidia驱动。
一、下载 官网下载tar版本的文件。
trt官网只更新到了支持ubuntu18.04，但是这个版本是可以在ubuntu20.04上使用的。
二、安装 下载之后解压：
tar -xzvf c-7.2.3.4.Ubuntu-18.04.x86_64-gnu.cuda-11.1.cudnn8.1.tar.gz

得到tensorRT文件夹，进入其中的python目录，里面提供了多种版本的tensorrt可供安装。 ubuntu20安装tensorRT
文章图片

建议在虚拟环境中安装。
conda create -n name python=3.7.0

注意：如果想要在python3.7中使用tensorRT，那么这里一定要写 3.7.0，否则可能报错，很坑。同理，想用3.6就写 3.6.0
创建完之后conda activate name进入新环境。开始安装。
这里就安装cp37的版本，使用命令：
pip install tensorrt-7.2.3.4-cp37-none-linux_x86_64.whl

可能的报错：
ERROR: tensorrt-7.2.3.4-cp37-none-linux_x86_64.whl is not a supported wheel on this platform原因：python版本不一致（就是刚刚说的问题）解决办法：确认anaconda环境中的python版本是否与要安装的tensorrt的python版本一致 找不到.so.7文件原因：.so.7文件在解压完的源文件中的lib文件夹里，需要将它添加到环境变量中解决办法： 1. 打开：sudo gedit ~/.bashrc 2. 在最后添加： export LD_LIBRARY_PATH="/tensorRT解压路径/lib:$LD_LIBRARY_PATH" 3. 刷新：source ~/.bashrc

安装完成后进入python输入tensorRT即可验证tensorRT是否安装成功 ubuntu20安装tensorRT
文章图片

不报错即成功.
接着进入tensorRT下的uff文件夹安装whl（tensorflow会用到）
pip install uff-0.6.9-py2.py3-none-any.whl

【ubuntu20安装tensorRT】至此，Ubuntu与TensorRT安装成功！
https://zhuanlan.zhihu.com/p/159591904

    推荐阅读


    上一篇：使用numpy计算两个框的iou

    下一篇：腾讯云运维工程师TCA证书能够提升我们职场竞争力