Skip to content

Latest commit

 

History

History
34 lines (29 loc) · 1.42 KB

README.md

File metadata and controls

34 lines (29 loc) · 1.42 KB

Onnx2trt Installation

These steps were successful installed on Jetson AGX device.

Command:

  1. 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
    
  2. Clone others submodules.

    git submodule update --init --recursive
    
  3. Cmake it.

    *This command is for Jetson.

    cmake . -DCUDA_INCLUDE_DIRS=/usr/local/cuda/include -DTENSORRT_ROOT=/usr/src/tensorrt -DGPU_ARCHS="53"
    
  4. Make and make insall.

    make 
    sudo make install
    

Reference