tensorrt_tutorial

sample 编译

https://github.com/NVIDIA/TensorRT/tree/release/8.0?tab=readme-ov-file#prerequisites

  1. github 上下载对应仓库，然后根据对应的 cuda 版本选择对应的tag进行下载。
    比如选择 release/8.0，如果有 docker 环境，可以利用官方提供的 docker 镜像进行；裸机的话需要安装cuda、cudnn、tensorrt 三件套

找到开发机中对应的 tensorrt lib 的路径，export TRT_LIBPATH=/usr/lib/aarch64-linux-gnu/

  1. 拉取三方依赖库的repo：git submodule update --init --recursive

一般需要 vpn 才能下载下来

  1. release/8.0版本需要修改两个地方

在 CMakeLists.txt中

设置nvcc 路径：set(CMAKE_CUDA_COMPILER /usr/local/cuda-10.2/bin/nvcc)
设置cub 的路径：set(CUB_ROOT_DIR ${CMAKE_CURRENT_SOURCE_DIR}/third_party/cub CACHE STRING "directory of CUB installation")

本博客所有文章除特别声明外，均采用 CC BY-SA 4.0 协议 ，转载请注明出处！