TensorRT部署

下載

在 https://developer.nvidia.com/nvidia-tensorrt-7x-download 選擇對應設備型號
LINUX查看CUDA版本 cat /usr/local/cuda/version.txt
LUNUX查看CUDNN版本 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
下載對應Tar文件

安裝

tar -xzvf TensorRT-XXX.tar.gz

然后設置路徑

nano ~/.bashrc
# 加入以下
export LD_LIBRARY_PATH=TensorRT解壓路徑/lib:$LD_LIBRARY_PATH
# 然后生效
source ~./bashrc

安裝Python接口

cd TensorRT-7.0.0.11/python
sudo pip3 install tensorrt-*-cp3x-none-linux_x86_64.whl
cd ../uff
sudo pip3 install uff-0.6.5-py2.py3-none-any.whl
cd ../graphsurgeon
sudo pip3 install graphsurgeon-0.4.1-py2.py3-none-any.whl

驗證安裝