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

5a2a01b799790767d94e02879413d7ad.png

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

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

windows10安装ubuntu双系统教程(绝对史上最详细) - 不妨不妨,来日方长 - 博客园​www.cnblogs.com
8df30c7dae00715a3c04bc402444a268.png

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

当安装Ubuntu的时候,注意以下选择:

6b4d736ab4fe5a2d60b63dc5ff4e8960.png

准备事项

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

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

db47914fa9d54509a9bb1f8d0282308d.png

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

be3137a407fdf3fd61c94f623f0252bb.png

至此,双系统安装成功!

接下来,就是配置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的开发者账号。

9f13258a8811d5ae8a08b54cd645a2b4.png
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

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

999dbde54df3185c8cfb53f964521186.png
  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

2c2f91bf8bbff535a78bc70419de840f.png

62afee6fff0c1dde472d96a29b672468.png

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

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 

c80f765342f0f904352e7354c4b4de1e.png

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

86b237ac6fd7977aa54d266c27bda4b0.png

不报错即成功

接着进入uff文件夹安装whl

bcf97fd31b40573224b57603485648ed.png

至此,Ubuntu与TensorRT安装成功!

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值