关于5090安装tensorrt(python api)的过程

前提条件

硬件5090

ubuntu24.04

cuda版本12.8

找到适配的tensorrt版本

Nvidia官网

完事了之后找到对应版本tar安装包

tar -xvzf tensorrt-你的安装包.tar

然后记得将路径加入到环境变量中

#在这里插入代码片
gedit ~/.bashrc
# 添加
export PATH=/PATH/To/TensorRT-你的按安装包/bin:${PATH}
export LD_LIBRARY_PATH=/PATH/TO/TensorRT-你的安装包/lib:${LD_LIBRARY_PATH}
# 更新
source ~/.bashrc
`

添加python api

在安装完了之后

打开

/path/to/TensorRT-10.10.0.31/python

打开这文件夹，有很多tensorrt-python的whl文件，直接在你的python环境中pip安装即可

例如你的python环境为3.11,直接安装

pip install tensorrt-你的版本-cp311-none-linux_x86_64.whl

然后就可以import tensorrt as trt了