TensorRT在Ubuntu20.04下的安装

一：去官网下载tar文件

根据安装环境下cuda的版本号，在官网中下载对应版本的tar包。

二：安装TensorRT

#解压tar包
sudo tar -xzvf TensorRT-8.5.1.7.Linux.x86_64-gnu.cuda-11.8.cudnn8.6.tar.gz
#添加环境变量
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/rjg/下载/TensorRT-8.5.1.7/lib （'注意/home/rjg/下载'是解压tar包的地址）
#安装tensorrt
cd TensorRT-8.5.1.7/python
pip install tensorrt-8.5.1.7-cp38-none-linux_x86_64.whl
#验证tensorrt安装是否成功，若正确应输出版本号: 8.5.1.7
python -c "import tensorrt;print(tensorrt.__version__)"
#安装uff
cd ../uff/
pip install uff-0.6.9-py2.py3-none-any.whl
#安装graphsurgeon
cd ../graphsurgeon/
pip install install graphsurgeon-0.4.6-py2.py3-none-any.whl

安装流程结束，测试图如下：