用TensorRT針對AArch64使用者的交叉編譯示例

wujianming_110117發表於2020-11-04

用TensorRT針對AArch64使用者的交叉編譯示例

以下介紹如何在x86_64linux下為AArch64 QNX和Linux平臺交叉編譯TensorRT示例。

2.1. Prerequisites

本節提供分步說明,以確保滿足交叉編譯的最低要求。

Procedure

  1. 為對應的目標安裝CUDA跨平臺工具包,並設定環境變數CUDA_INSTALL_DIR。

$ export CUDA_INSTALL_DIR=“your cuda install dir”

Where CUDA_INSTALL_DIR is set to /usr/local/cuda by default.

  1. 為相應的目標安裝cudn跨平臺庫,並設定環境變數CUDNN_INSTALL_DIR。

$ export CUDNN_INSTALL_DIR=“your cudnn install dir”

Where CUDNN_INSTALL_DIR is set to CUDA_INSTALL_DIR by default.

  1. 為相應的目標安裝TensorRT交叉編譯Debian包。

注意:如果您使用目標平臺的tar檔案版本,那麼您可以安全地跳過這一步。tar檔案版本已經包含了交叉編譯庫,因此不需要額外的包。

QNX AArch64

libnvinfer-dev-cross-qnx,
libnvinfer7-cross-qnx

Linux AArch64

libnvinfer-dev-cross-aarch64,
libnvinfer7-cross-aarch64

2.2. Building Samples For QNX AArch64

本節提供為QNX使用者構建示例的分步說明。

下載QNX工具鏈並匯出以下環境變數。

$ export QNX_HOST=/path/to/your/qnx/toolchains/host/linux/x86_64

$ export
QNX_TARGET=/path/to/your/qnx/toolchain/target/qnx7

通過釋出以下命令構建示例:

$ cd /path/to/TensorRT/samples

$ make TARGET=qnx

2.3. Building Samples For Linux AArch64

本節提供為Linux使用者構建示例的分步說明。

安裝相應的GCC編譯器aarch64-linux-gnu-g++。在Ubuntu中,可以通過以下方式安裝:

$ sudo apt-get install g+±aarch64-linux-gnu

通過釋出以下命令構建示例:

$ cd /path/to/TensorRT/samples

$ make TARGET=aarch64

相關文章