git clone https://github.com/YanBC/nnhelper.git
cd nnhelper
make
After installation, the following tools should be available from command line:
info_onnx
: show onnx model inputs and outputsinfo_trt
: show tensorrt model inputs and outputslist_trt_plugins
: list all available tensorrt pluginsdevice_query
: list properties of CUDA enabled GPUs
By default, these tools are all installed under ~/.local/bin/