Libraries & Packages/tensorRT3 [TensorRT] pytorch network를 c++에서 inference * Enviornment Jetson Orin NX Jetpack 5.1 [L4T 35.2.1] Ubuntu 20.04 cuda 11.4 1. python에서 network를 onnx로 변환 # python neural_network = your_network().to(device) neural_network.load_state_dict(torch.load("your_weight")) x = torch.randn("your_input_shape").to(device) torch.onnx.export(neural_network, x, "your_model_path.onnx", export_params=True, opset_version=10, do_constant_folding=True, input_nam.. 2023. 11. 17. [tensorRT] CMakelists.txt 에 tensorRT 추가 CMakelists.txt에 다음 내용추가 find_package(CUDA REQUIRED) list(APPEND PLUGINS "nvinfer") list(APPEND PLUGINS "nvonnxparser") list(APPEND PLUGINS "nvparsers") foreach(libName ${PLUGINS}) find_library(${libName}_lib NAMES ${libName} "/usr" PATH_SUFFIXES lib) list(APPEND PLUGIN_LIBS "${${libName}_lib}") endforeach() include_directories( ${CUDA_INCLUDE_DIRS} ${CUDNN_ROOT_DIR}/include ) include_directories.. 2023. 11. 16. trtexec: command not found 에러 * environment Jetson Orin NX Jetpack 5.1 * error 터미널에 trtexec 실행 시 trtexec: command not found 에러 발생 * 해결방법 bash파일에 다음 줄 추가 alias trtexec="/usr/src/tensorrt/bin/trtexec" * 참고 https://forums.developer.nvidia.com/t/bash-trtexec-command-not-found/127302 Bash: trtexec: command not found Description I tried to build trtexec in .../TensorRT/samples. I followed this git link for building the sample but .. 2023. 11. 16. 이전 1 다음