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

最新推荐文章于 2024-07-25 18:26:39 发布

原创

最新推荐文章于 2024-07-25 18:26:39 发布 · 924 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#tx2 #TensorRT #Jetpack #Samples #TensorRT6.0

本文详细介绍如何在JetPack4.3环境下轻松集成TensorRT，包括C++和Python版本的安装配置，以及如何编译和使用C++样例。特别针对TX2平台的Arm架构进行了说明，分享了解决protobuf下载问题的方法。

不得不说，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

注意三点