Ubuntu18.04安装TensorRT5.1.5

前言

深度学习中使用到了TesnsorRT,安装过程整理如下

安装TensorRT5.1.5

首先下载对应版本的TensorRT。如TensorRT-5.1.5.0.Ubuntu-18.04.2.x86_64-gnu.cuda-10.0.cudnn7.5.tar.gz,压缩包的名称中就包含了cuda以及cudnn的版本,对应即可。
按照下面命令安装即可

$ tar -zxvf TensorRT-5.1.5.0.Ubuntu-18.04.2.x86_64-gnu.cuda-10.0.cudnn7.5.tar.gz 
$ cd TensorRT-5.1.5.0/python/
$ sudo apt install python-pip
$ pip install tensorrt-5.1.5.0-cp27-none-linux_x86_64.whl 
$ cd ../graphsurgeon/
$ pip install graphsurgeon-0.4.1-py2.py3-none-any.whl 

之后要记得配置环境

sudo gedit ~/.bashrc

export LD_LIBRARY_PATH=/home/xxxx/TensorRT-5.1.5.0/lib:$LD_LIBRARY_PATH
将上述语句添加到环境变量,xxxx替换为自己的目录名称
,保存退出后使其生效:

source ~/.bashrc
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。