TX2编译TensorRT的C++ samples ——Jetpack4.3

不得不说,JetPack 4.3实在是太方便了!

1. Where is TensorRT

1.1 C++版本

与手动安装tensorRT不同,JetPack自动将TensorRT c++的lib中的文件保存在了

/usr/lib/aarch64-linux-gnu/

1.2 Python版本

import tensorrt
tensorrt.__version__
#6.0.1.10

2.编译C++ Sample

git clone -b release/6.0 https://github.com/nvidia/TensorRT TensorRT
cd TensorRT
git submodule update --init --recursive
export TRT_SOURCE=`pwd`
mkdir -p build && cd build
cmake .. -DTRT_LIB_DIR=/usr/lib/aarch64-linux-gnu/ -DTRT_BIN_DIR=`pwd`/out
make -j$(nproc) TARGET=aarch64

注意三点

  • git clone时要选择6.0
  • 修改TensorRT lib的路径
  • 编译时选择TARGET为aarch64,因为TX2为Arm架构的.
    参考网址

3.使用C++ Sample

注意使用out/里的例子时,要先将so文件复制到TensorRT lib路径下,即

cd out/
sudo cp *.so* /usr/lib/aarch64-linux-gnu

4.可能遇到的问题

可能会遇到protobuf下载时网络不响应的问题,原因在于curl不能识别https://开头的网址,需要重新编译curl.
办法见TX2编译安装curl,使其可以解析https

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章