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