TensorRT在ubuntu18.04的安装

摘要:
安装TensorRT前需要安装Cuda和cudnn,安装步骤可以参考ubuntu安装cuda和cudnn。
配置环境
ubuntu 18.04 ,cuda 10.1 , cudnn 7.5 , python3.5 ;
安装步骤
1.首先下载tar版本的安装包,下载地址( https://developer.nvidia.com/nvidia-tensorrt-5x-download)需要登陆NVIDIA。
安装TensorRT前需要安装Cuda和cudnn,安装步骤可以参考 ubuntu安装cuda和cudnn。
2. 打开下载的TensorRT所在路径,解压下载的tar文件:
$ tar -xzvf TensorRT-XXX.tar.gz

3. 解压添加环境变量

$ vim ~/.bashrc # 打开环境变量文件
# 将下面三个环境变量写入环境变量文件并保存
export LD_LIBRARY_PATH=TensorRT解压路径/lib:$LD_LIBRARY_PATH
export CUDA_INSTALL_DIR=/usr/local/cuda-9.0
export CUDNN_INSTALL_DIR=/usr/local/cuda-9.0
# 使刚刚修改的环境变量文件生效
$ source ~/.bashrc

4. 下面是安装Python的TensorRT包:进到解压的TensorRT目录下的Python目录:

# 对于python2
$ sudo pip2 install tensorrt-XXX-cp27-cp27mu-linux_x86_64.whl
# 对于python3
$ sudo pip3 install tensorrt-XXX-cp35-cp35m-linux_x86_64.whl

5. 测试TensorRT是否安装成功,进入Python编辑器加载tensorrt:

>>> import tensorrt 
>>> tensorrt.__version__

6. 错误分析: import tensorrt 时出现:

ImportError: libcudnn.so.7: cannot open shared object file: No such file or directory

解决方法:

cd /usr/local/cuda/lib64
sudo rm -rf libcudnn.so libcudnn.so.7  #删除原有版本号,版本号在cudnn/lib64中查询
sudo ln -s libcudnn.so.7.0.5 libcudnn.so.7 #生成软连接,注意自己下载的版本号
sudo ln -s libcudnn.so.7 libcudnn.so
sudo ldconfig #立即生效

免责声明:文章转载自《TensorRT在ubuntu18.04的安装》仅用于学习参考。如对内容有疑问,请及时联系本站处理。

上篇windows下vue项目启动步骤可爱猫+python——定制化微信机器人下篇

宿迁高防,2C2G15M,22元/月;香港BGP,2C5G5M,25元/月 雨云优惠码:MjYwNzM=

相关文章

Linux环境下搭建go开发环境

一、环境介绍 Linux环境:阿里云,Linux镜像是centos8.2 64位 本机电脑:是win10操作系统 二、Linux下golang环境搭建 首先通过Xshell6连接上阿里云Linux操作系统    下载go语言开发环境,地址是https://golang.google.cn/dl/,可以根据自己的情况选择相应的版本,我选择的是go1.15....

openstack的glance、nova、cinder使用ceph做后端存储

块设备与 OPENSTACK 通过 libvirt 你可以把 Ceph 块设备用于 OpenStack ,它配置了 QEMU 到 librbd 的接口。 Ceph 把块设备映像条带化为对象并分布到集群中,这意味着大容量的 Ceph 块设备映像其性能会比独立服务器更好。要把 Ceph 块设备用于 OpenStack ,必须先安装 QEMU 、 libvirt...

小团队开发管理工具:gitlab+redmine+testlink+jenkins

由于工作需要,需要为团队搭建一个高效可用的开发管理平台。现在可用的开发管理工具很多开源的、商业的,网上也有很多博客和文章。经过2周的学习比较,再结合自己的项目特点,最后选定工具集:gitlab+redmine+testlink+jenkins。 项目管理和bug跟踪工具:redmine。整个项目组一直使用的管理工具,这次只是做了升级(2.4.0) 代码管...

linux系统下安装jdk教程

一、首先下载linux版本jdk jdk官网:http://www.oracle.com/technetwork/java/javase/downloads/index.html 根据自己的需求,下载不同版本的jdk 2.将下载好的jdk压缩包,通过ftp上传到linux系统的当前用户下,我当前登录的用户为root用户 3.将上传后的jdk,解压到/...

(转载)linux中shell变量

(转载)http://blog.csdn.net/zahuopuboss/article/details/8633891 为使shell编程更有效,系统提供了一些shell变量。shell变量可以保存诸如路径名、文件名或者一个数字这样的变量名。shell将其中任何设置都看做文本字符串。有两种变量,本地和环境。严格地说可以有4种,但其余两种是只读的,可以认为...

前用户sudo免密码

  Ubuntu 设置当前用户sudo免密码 方法1 # 备份 /etc/sudoers sudo cp /etc/sudoers . #打开 /etc/sudoers sudo visudo # 在文件末尾加入 linuxidc ALL=NOPASSWD:ALL 方法2 1. 备份sudo文件 sudo cp /etc/sudoers ....