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

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