Linux,Unix,BSD/Jetson

[Jetson] NVIDIA TensorRT 버전 확인하는 방법

채윤아빠 2022. 7. 8. 07:47
728x90
반응형

개요

NVIDIA TensorRT를 이용하던 중, 버전이 일치하지 않으므로, 다시 빌드 하라는 오류를 만났습니다.

ERROR: [TRT]: 6: The engine plan file is not compatible with this version of TensorRT, expecting library version 8.4.0.9 got 8.4.0.11, please rebuild.
ERROR: [TRT]: 4: [runtime.cpp::deserializeCudaEngine::49] Error Code 4: Internal Error (Engine deserialization failed.)

그래서 현재 이용중인 TensorRT의 버전을 확인하는 방법을 정리해 둡니다.


dpkg

다음과 같이 "nvinfer" 또는 "TensorRT"로 걸러서 버전을 확인할 수 있습니다.

hbesthee@my-desktop:~$ dpkg -l | grep nvinfer
ii  libnvinfer-bin            8.4.0-1+cuda11.4    arm64    TensorRT binaries
ii  libnvinfer-dev            8.4.0-1+cuda11.4    arm64    TensorRT development libraries and headers
ii  libnvinfer-doc            8.4.0-1+cuda11.4    all      TensorRT documentation
ii  libnvinfer-plugin-dev     8.4.0-1+cuda11.4    arm64    TensorRT plugin libraries
ii  libnvinfer-plugin8        8.4.0-1+cuda11.4    arm64    TensorRT plugin libraries
ii  libnvinfer-samples        8.4.0-1+cuda11.4    all      TensorRT samples
ii  libnvinfer8               8.4.0-1+cuda11.4    arm64    TensorRT runtime libraries
ii  python3-libnvinfer        8.4.0-1+cuda11.4    arm64    Python 3 bindings for TensorRT
ii  python3-libnvinfer-dev    8.4.0-1+cuda11.4    arm64    Python 3 development package for TensorRT

hbesthee@my-desktop:~$ dpkg -l | grep TensorRT
ii  libnvinfer-bin            8.4.0-1+cuda11.4    arm64    TensorRT binaries
ii  libnvinfer-plugin8        8.4.0-1+cuda11.4    arm64    TensorRT plugin libraries
ii  libnvinfer8               8.4.0-1+cuda11.4    arm64    TensorRT runtime libraries
ii  libnvonnxparsers8         8.4.0-1+cuda11.4    arm64    TensorRT ONNX libraries
ii  libnvparsers8             8.4.0-1+cuda11.4    arm64    TensorRT parsers libraries
ii  python3-libnvinfer        8.4.0-1+cuda11.4    arm64    Python 3 bindings for TensorRT

nm

하지만, 빌드 번호가 일치하지 않는 경우라면, 다음과 같은 방법으로 빌드 번호까지 확인해야 합니다.

hbesthee@my-desktop:~$ whereis libnvinfer.so
libnvinfer: /usr/lib/aarch64-linux-gnu/libnvinfer.so
hbesthee@my-desktop:~$ nm -D /usr/lib/aarch64-linux-gnu/libnvinfer.sogrep tensorrt_version
000000001bd4a618 B tensorrt_version_8_4_0_9

참고자료

"How to check my TensorRT version":https://forums.developer.nvidia.com/t/how-to-check-my-tensorrt-version/56374/10