1. 首先安装好cuda10.2
2.官网下载对应的cuda10.2版本的tensorrt
官网网址:
https://developer.nvidia.com/nvidia-tensorrt-8x-download
3.解压
4.添加环境变量
sudo gedit ~/.bashrc
export LD_LIBRARY_PATH=/home/xmy/TensorRT-8.4.3.1/lib:$LD_LIBRARY_PATH
export LIBRARY_PATH=/home/xmy/TensorRT-8.4.3.1/lib::$LIBRARY_PATH
source ~/.bashrc
5. 进入到tensorrt文件下的python文件选择对应的python版本
cd ./TensorRT-8.4.3.1/python
6. 根据python版本建立自己的虚拟环境并激活虚拟环境
conda activate tensorrt
pip install tensorrt-8.4.3.1-cp37-none-linux_x86_64.whl
7. 安装成功并验证
参考链接: