一 安装Ubuntu+配置环境+安装TensorRT

Posted 2021-08-30 mishidemudong

最近研究推理加速，涉及到TensorRT，加上之前一直也想搞个Ubuntu系统，所以就打算一步到位，双系统安排。

刻录U盘的前期步骤可以参考文章：

windows10安装ubuntu双系统教程（绝对史上最详细） - 不妨不妨，来日方长 - 博客园​www.cnblogs.com

这里注意前期不用分区，那些分区的步骤直接跳过，我自己最开始就是按照分区进行安装，一直有问题，所以这里不建议参考分区。

当安装Ubuntu的时候，注意以下选择：

准备事项

注：如果网络和空间匀速，建议选择“安装中下载更新”和“安装这个第三方软件”。

非常重要的一步，选择“安装Ubuntu 与 其他系统共存”：系统自动会分出挺大内存给Ubuntu

安装完成后再重启就是选择双系统的界面，这里根据电脑不同所展示画面会有不同

至此，双系统安装成功！

接下来，就是配置Anaconda + CUDA + CUDNN + TensorRT

这里说一个小技巧：先确定你要安装的TensorRT版本，再确定CUDA 以及CUDNN版本。

我就是一开始先装了CUDA 10.2 + CUDNN 8.0 ，之后安装TensorRT-7.0.0.11.Ubuntu-18.04.x86_64-gnu.cuda-10.2.cudnn7.6.tar.gz的时候发现匹配的CUDNN版本不对，又重新安装cudnn7.6，所以这里希望大家先确定要安装的TensorRT版本，再确定CUDA 以及CUDNN版本进行安装。

---

这里就直接跳到tensorRT安装部分。

下载安装包并解压

下载tar版本的安装包，同样需要登入NVIDIA的开发者账号。

tar -xzvf /Downloads/TensorRT-7.0.0.11.Ubuntu-18.04.x86_64-gnu.cuda-10.2.cudnn7.6.tar.gz

安装python的tensorRT包

进入tensorRT目录下的Python目录

cd /TensorRT-7.0.0.11/python
# 对于python3
$ sudo pip3 install tensorrt-7.0.0.11-cp37-none-linux_x86_64.whl

这里注意可能会遇到两个问题：

1. sudo: pip3 :command not found问题解决

sudo apt-get install python3-pip

2. ERROR: tensorrt-7.0.0.11-cp37-none-linux_x86_64.whl is not a supported wheel on this platform问题解决

原因是cp37即python=3.7.0,如果你的anaconda下的python版本不是3.7.0就会报错。所以这里我使用创建虚拟环境解决

conda create --name python370 python=3.7.0

虚拟环境创建成功后进行激活再安装：

conda activate python370
cd /TensorRT-7.0.0.11/python
python -m pip install tensorrt-7.0.0.11-cp37-none-linux_x86_64.whl 

进入python输入tensorRT即可验证tensorRT是否安装成功

不报错即成功

接着进入uff文件夹安装whl

至此，Ubuntu与TensorRT安装成功！

下一篇打算说一下Windows下安装TensorRT，以及涉及到多版本CUDA以及虚拟环境的对应及调用。