不得不說,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