1、下載TensorRT
下載地址：
https://developer.nvidia.com/nvidia-tensorrt-download
需要登錄nvidia官網，還得回答問題，隨便填即可
建議選擇壓縮包版本
我選擇的是TensorRT8.x
2、配置
解壓后進入TensorRT根目錄：
（1）環境變量
vi ~/.bashrc
在文件末尾添加一行代碼：
export LD_LIBRARY_PATH=/opt/TensorRT-8_0_1_6/lib:$LD_LIBRARY_PATH
關閉保存
source ~/.bashrc
（2）復制文件到系統路徑
把TensorRT根目錄中的/lib/下面的文件復制到 /usr/lib/下，
把TensorRT根目錄中的/include/下面的文件復制到 /usr/include/下
3、安裝python包
進入TensorRT根目錄下的python/目錄下
ll
可以看到多個版本的python包。
因為我之前安裝的是python3.10版，所以選擇安裝文件tensorrt-8.4.1.5-cp310-none-linux_x86_64.whl
執行安裝命令：
pip install --force-reinstall tensorrt-8.4.1.5-cp310-none-linux_x86_64.whl
4、測試
執行命令:
python3
進入python控制臺
執行python 命令：
import tensorrt
可以看到沒有報錯
注意：
如果第2節的步驟（2）中，沒有把TensorRT根目錄中的/lib/下面的文件復制到 /usr/lib/下，則此處報錯：
libnvinfer.so.8: cannot open shared object file: No such file or directory
執行python命令：
tensorrt.__version__
可以查看到tensorrt版本號：
‘8.4.1.5’
O了！
posted on 2022-07-28 15:05
小王 閱讀(2111)
評論(0) 編輯 收藏 引用 所屬分類:
編譯 、
TensorRT