下载对应版本的压缩包,需要先登录,地址是tensorRT下载地址

进入列表后点击找到对应版本; 8.x下载列表

Ubuntu使用tar包方式安装tensorRT_压缩包

下载后解压 添加环境变量

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/TensorRT-8.6.1.6/lib

复制到系统目录,TensorRT路径下

sudo cp -r ./lib/* /usr/lib
sudo cp -r ./include/* /usr/include

到Python目录下,安装TensorRT

cd /opt/TensorRT-8.6.1.6/python
pip install tensorrt-8.6.1-cp311-none-linux_x86_64.whl

到uff目录下,安装UFF,支持tensorflow模型转化

cd /opt/TensorRT-8.6.1.6/uff
pip install uff-0.6.9-py2.py3-none-any.whl

到graphsurgeon目录下,安装graphsurgeon,支持自定义结构

cd /opt/TensorRT-8.6.1.6/graphsurgeon
pip install graphsurgeon-0.4.6-py2.py3-none-any.whl

测试一下. 进入Python

import tensorrt as trt
print(trt.__version__)