These steps were successful installed on Jetson AGX device.
-
Clone the repository.
git clone https://github.com/onnx/onnx-tensorrt.git && cd onnx-tensorrt
BTW, Current version is supported by TensorRT 7.0, if your TensorRT is not 7.0, you should change to others branch.
*This part is for AGX that it needs a lower version.
git checkout 6.0-full-dims
-
Clone others submodules.
git submodule update --init --recursive
-
Cmake it.
*This command is for Jetson.
cmake . -DCUDA_INCLUDE_DIRS=/usr/local/cuda/include -DTENSORRT_ROOT=/usr/src/tensorrt -DGPU_ARCHS="53"
-
Make and make insall.
make sudo make install