下载对应版本的压缩包,需要先登录,地址是tensorRT下载地址
进入列表后点击找到对应版本; 8.x下载列表
下载后解压 添加环境变量
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__)