关于深度学习:ubuntu20安装tensorRT

7次阅读

共计 1075 个字符,预计需要花费 3 分钟才能阅读完成。

大体来说装置的办法有两种:

  • 用 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 可供装置。

倡议在虚拟环境中装置。

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 是否装置胜利

不报错即胜利.

接着进入 tensorRT 下的 uff 文件夹装置 whl(tensorflow 会用到)

pip install uff-0.6.9-py2.py3-none-any.whl 

至此,Ubuntu 与 TensorRT 装置胜利!


https://zhuanlan.zhihu.com/p/159591904

正文完
 0