【ubuntu】安装tensorRT

2023-11-03

**tensorRT官方安装文档TensorRT3-Installation-Guide-RC.pdf位于tensorRT下载页面**

1,tensorRT下载,将tensorRT下载到想要安装的目录

https://developer.nvidia.com/rdp/form/tensorrt3RC-download-survey

注:tensorRT官方提供两种安装版本tar和deb安装,但用deb安装多次均未成功,以下只列出了tar安装教程

2,需要预先安装CUDA Toolkit8.0 or9.0 cuDNN7.0 python2/3

CUDA安装教程:http://blog.csdn.net/xll_bit/article/details/78304722

cudnn安装教程:http://blog.csdn.net/xll_bit/article/details/78373576

3,解压tar文件,解压完会出现TensorRT.-3.0.0,这个目录下面会包含lib,bin等文件

tar -xzvf TensorRT-3.0.0.Ubuntu-16.04.3.cuda-9.0.tar.gz(此命令根据下载位置不同有可能需要sudo权限)

4,添加环境变量

vim ~/.bashrc 

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:<eg:TensorRT-3.0.0/lib>

source ~/.bashrc

5,安装python tensorRT包

$ cd TensorRT-3.0.0/python #转到解压后的python目录

对于python2.7

$ sudo pip2 install tensorrt-3.0.0-cp27-cp27mu-linux_x86_64.whl

对于python3.5

$ sudo pip3 install tensorrt-3.0.0-cp35-cp35m-linux_x86_64.whl

测试 tensorRT是否安装成功

$ which tensorrt
/usr/local/bin/tensorrt

**此步骤我在一台服务器上能运行通过,在另一台服务器上运行报错**

错误记录:

In file included from src/cpp/cuda.cpp:1:0:
    src/cpp/cuda.hpp:14:18: fatal error: cuda.h: No such file or directory
    compilation terminated.
    error: command 'x86_64-linux-gnu-gcc' failed with exit status 1

显示是找不到cuda.h,根据网上分析是因为用了sudo之后环境变量用的是root的环境变量

具体可参考https://codeyarns.com/2015/07/31/pip-install-error-with-pycuda/

主要原因是安装pycuda时依赖cuda库,sudo之后找不到所以出错

解决方法:

vim /etc/profile.d/cuda.sh

添加:export PATH=/usr/local/cuda-8.0/bin:$PATH

$ sudo su -

$ pip2 install tensorrt-3.0.0-cp27-cp27mu-linux_x86_64.whl --trusted-host pypi.douban.com

$ exit

6,安装UFF包

转到uff目录下

sudo pip2 install uff-0.1.0rc0-py2.py3-none-any.whl

或 sudo pip3 install uff-0.1.0rc0-py2.py3-none-any.whl

测试:

$ which convert-to-uff
/usr/local/bin/convert-to-uff

7,复制lenet5.uff文件到相应目录

sudo cp TensorRT-3.0.0/data/mnist/lenet5.uff TensorRT-3.0.0/python/data/mnist/lenet5.uff

8,验证

cd TensorRT-3.0.0/samples/sampleMNIST

sudo make clean

sudo make

cd /TensorRT-3.0.0/bin(转到bin目录下面,make后的可执行文件在此目录下)

sudo ./sample_mnist

tensorrt2安装:

解压tar文件到指定目录

tar -jxvf TensorRT-2.1.2.x86_64.cuda-8.0-14-04.tar.bz2 -C /opt

添加环境变量:(~/.bashrc  or  /etc/profile)

export CUDA_INSTALL_DIR=/usr/local/cuda-8.0
export CUDNN_INSTALL_DIR=/usr/local/cuda-8.0
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/TensorRT-2.1.2/lib

验证同tensorrt3




本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

【ubuntu】安装tensorRT 的相关文章

随机推荐

  • [架构之路-182]-《软考-系统分析师》-19- 系统可靠性分析与设计 - 概览

    前言 可靠性工程是研究产品生命周期中故障的发生 发展规律 达到预防故障 消灭故 障 提高产品可用性的工程技术 信息系统的可靠性是指系统在满足一定条件的应用环境中能够正常工作的能力 可以按一般工程系统的可靠性标准进行定性评价 也可以通过平均无
  • 分享24个Python接单平台,有技术等于有收入!

    一 Python兼职种类 接私活刚学会python那会 就有认识的朋友介绍做一个网站的私活 当时接单赚了4K 后又自己接过开发网站后台接口 做数据处理等事情 都赚了一些 接私活指的是利用自己的技术 在业余时间搞定用户整块需求 对方在开始前预
  • Java自动化框架配置监听器生成测试报告邮件发送

    TestNG官网 https testng org doc documentation main html introduction TestNG博客 https www jianshu com p 8a729de618b6 一 接口自动化
  • JMETER接口测试_参数化和关联实现注册、登录和查询

    JMETER接口测试 参数化和关联 实现如下 1 注册接口 实现参数化 2 登录接口 关联用第一步中的用户和密码 3 博文查询接口 关联登录接口返回的token和userid 1 添加Thread Group 2 添加HTTP Reques
  • NameNode: Permission denied&无法启动Hadoop解决方法

    NameNode Permission denied publickey gssapi keyex gssapi with mic password 就是这个原因 这个问题的出现主要是因为没有给authorized keys授权 解决方法如
  • VMware虚拟机安装MacOS Big Sur

    之前完善了vm安装Windows系统的教程 今天给大家分享一个vm安装MacOS的教程 我们今天用macOS Big Sur版本来做教程演示 注 使用VMware安装MacOS哪怕配置给的高也会出现体验上的不佳 大家可以尽可能调高适当的配置
  • ElasticSearch 双数据中心建设在新网银行的实践

    本文公众号读者飞熊的投稿 本文主要讲述了ElasticSearch 双数据中心建设在新网银行的实践 作者简介 飞熊 目前就职于新网银行大数据中心 主要从事大数据实时计算和平台开发相关工作 对Flink Spark 以及ElasticSear
  • goland语法面试题

    文章目录 1 关于 switch 语句 下面说法正确的是 2 下面代码能编译通过吗 可以的话 输出什么 3 interface 是可以指向任意对象的 Any 类型 是否正确 4 下面的代码有什么问题 1 关于 switch 语句 下面说法正
  • Unity5热更新ILRuntime 使用 Protobuf3.0

    Unity5热更新ILRuntime 使用 Protobuf3 0 须知 1 pb3官方用到了C 很多的新语法 所以在unity主工程中直接撸码是不可以的 还好github上面有同僚作了framework35版的 2 ILrt中的类目前是不
  • R语言与面向对象的编程(3):R6类

    专注系列化 高质量的R语言教程 本号已支持快捷转载 无需白名单即可转载 本系列将介绍R语言中三个与面向对象的编程 Object Oriented Programming OOP 相关的工具包 proto R6和基础包methods 这是一个
  • python中,@和-> 代表什么?

    今天把代码放到Hadoop平台时调试代码的时候报错 但是在本地测试并没有什么问题 然后可查看了下代码 报错的地方这么定义的 看到这个符号觉得很奇怪 因为在Python中确实没见过这个符号 后来查了一下 参考这个博主写的 https blog
  • noip2008 火柴棒等式 (暴力枚举)

    P1496火柴棒等式 Accepted 标签 搜索 NOIP提高组2008 描述 给你n根火柴棍 你可以拼出多少个形如 A B C 的等式 等式中的A B C是用火柴棍拼出的整数 若该数非零 则最高位不能是0 用火柴棍拼数字0 9的拼法如图
  • 算法基础\BFS\DFS

    1 200 岛屿数量 题目描述 给你一个由 1 陆地 和 0 水 组成的的二维网格 请你计算网格中岛屿的数量 岛屿总是被水包围 并且每座岛屿只能由水平方向和 或竖直方向上相邻的陆地连接形成 此外 你可以假设该网格的四条边均被水包围 示例 示
  • 41.cuBLAS开发指南中文版--cuBLAS中的Level-2gemvBatched()

    2 6 24 cublas
  • numpy 三维矩阵下采样小技巧

    问题描述 我们有一个 160 192 224 的三维矩阵 由于其过大 跑神经网络时显存不够 此时我们的一个思路就是对其进行一个简单的下采样 即没3x3x3的小方格里只取其中间的那个数 现在的问题是如何快速高效地实现这个计算 而不是使用多个f
  • 爬虫代理IP池怎么来的,可能遇到哪些问题,怎么解决

    目录 前言 一 代理IP对爬虫工作的重要性 二 代理IP池从哪里来 三 爬虫工作中可能会遇到哪些问题 四 怎么解决遇到的问题 总结 前言 爬虫工作离不开代理IP的支持 代理IP在爬虫工作中发挥重要的作用 但爬虫代理IP池从哪里来呢 爬虫工作
  • 口令破解(概述、暴力破解、字典破解、Hydra)

    文章目录 口令破解 概述 口令安全现状 破解方式 暴力破解 字典破解 Hydra quarkspwdump 口令破解 概述 现在很多地方都以用户名 账号 和口令 密码 作为鉴权的方式 口令 密码 就意味着访问权限 口令 密码 就相当于进入家
  • sata接口_无线网卡M.2(ngff) keyA/E接口扩展sata硬盘接口,黑群辉NAS系统

    miniPC普及 主板小型化后 PCIE接口数量减少到1 2个 甚至没有 sata接口数量也大幅减少 这样的主板对于想使用多硬盘 特别是要黑群辉NAS系统的 是不适合的 好在某些主板有无线网卡M 2 ngff 接口 为扩展sata存留了希望
  • 华为手机助手上架流程_2019年各大安卓应用商店上架经验,含流程,物料,方法,建议收藏...

    注册应用商店账号 申请应用商店上架是APP推广的第一步 这一步没做好 会延迟甚至耽误后续的工作 因此 做好上架工作尤为重要 今天姑婆根据我们自己APP上架的经验进行了整理 分享给大家 希望对大家有所帮助 一 安卓应用商店格局 据调查显示在中
  • 【ubuntu】安装tensorRT

    tensorRT官方安装文档TensorRT3 Installation Guide RC pdf位于tensorRT下载页面 1 tensorRT下载 将tensorRT下载到想要安装的目录 https developer nvidia