天天看点

ubuntu安装TensorRT

ubuntu18.04+cuda11.2+cudnn8.0.5+Tensorrt7.2.2.3安装

环境

系统:ubuntu 18.04

编译器:gcc 7.5

如果之前安装过deb格式的tensorrt,需要卸载

如果用pip安装过对应的包(根据自己安装过的包卸载)

由于之前安装过其他版本的tensorrt,安装新的tensorrt时会默认安装key中最新版本的tensorrt,因此想要安装老版本的tensorrt需要删除其他版本的key。

查看key

查看key列表

删除key

删除安装包

官网教程

cuda和cudnn也要是deb安装

验证

进入<code>/var/nv-tensorrt-repo-ubuntu1804-cuda11.1-trt7.2.3.4-ga-20210226</code>

安装剩下的包,eg:uff

如果要安装python接口,下载压缩包版本的tensorrt,在解压后的目录中通过pip安装对应的whl

<code>sudo pip3 install TensorRT-${version}/python/tensorrt-*-cp3x-none-linux_x86_64.whl</code>

可能会遇到部分包缺失依赖

安装对应的依赖

如果提示数据不存在,进入<code>data/mnist</code>目录,运行<code>download_pgms.py</code>下载数据

解压压缩包

写入路径

如果要安装python接口,在解压后的目录中通过pip安装对应的whl