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

安装流程结束,测试图如下:

TensorRT在Ubuntu20.04下的安装_第1张图片

你可能感兴趣的:(模型部署,python,linux,开发语言)