TensorRT开源组件编译环境配置

最新推荐文章于 2024-07-23 09:23:56 发布

weixin_45415546

最新推荐文章于 2024-07-23 09:23:56 发布

阅读量363

点赞数

分类专栏： TensorRT 文章标签：深度学习

本文链接：https://blog.csdn.net/weixin_45415546/article/details/103071798

版权

TensorRT 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

本文详细介绍了如何设置TensorRT环境变量，包括TRT_RELEASE路径、更新LD_LIBRARY_PATH以及使用cmake进行配置的具体命令。这对于在Linux环境下搭建深度学习推理平台至关重要。

摘要由CSDN通过智能技术生成

TensorRt环境变量

 export TRT_RELEASE=`pwd`/TensorRT
 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$TRT_RELEASE/lib

cmake -D CMAKE_CUDA_COMPILER=/usr/local/cuda-10.0/bin/nvcc -DTRT_LIB_DIR=$TRT_RELEASE/lib -DTRT_BIN_DIR=`pwd`/out .

 cmake .. -DTRT_LIB_DIR=$TRT_RELEASE/lib -DTRT_BIN_DIR=`pwd`/out

我的TensorRt环境变量

 export TRT_RELEASE=/usr/local/TensorRT-KL
 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$TRT_RELEASE/lib

cmake -D CMAKE_CUDA_COMPILER=/usr/local/cuda-10.0/bin/nvcc -DTRT_LIB_DIR=$TRT_RELEASE/lib -DTRT_BIN_DIR=`pwd`/out .

 cmake .. -DTRT_LIB_DIR=$TRT_RELEASE/lib -DTRT_BIN_DIR=`pwd`/out