tensorRT安装(提前编译opencv)

原创

于 2021-04-08 09:57:24 发布 · 540 阅读

CC 4.0 BY-SA版权

本文记录了在Ubuntu系统中，如何手动安装TensorRT的详细步骤，包括查询Ubuntu、CUDA和cuDNN版本，下载匹配的TensorRT tar包，以及安装后在Anaconda环境中测试模块导入。

注：这里仅记录用tar安装tensorRT的流程，注意的是解压版的tar需要我们手动去修改lib的路径，如果是使用安装版的，就不需要了，安装版需要root权限进行安装。
1.查询自己的ubuntu、CUDA、cuDNN版本：

# 查看cuda版本
cat /usr/local/cuda/version.txt
# 查看cuDNN版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
# 查看ubuntu版本
cat  /etc/issue

官网下载地址：https://developer.nvidia.com/nvidia-tensorrt-download

# 在home下将下载的压缩文件进行解压
tar xzvf TensorRT-6.0.1.5.Ubuntu-16.04.x86_64-gnu.cuda-10.1.cudnn7.6.ta