当前位置： 首页 > article >正文

linux版：TensorRT安装教程

article 2024/11/21 1:42:46

首先安装cuda和cudnn

查看安装的cuda版本

nvidia-smi

在这里插入图片描述

查看cuda电脑的版本，我的是11.8版本的

nvcc -V

在这里插入图片描述

查看cudnn版本，在cuda/cuda118/include路径下

cat cudnn_version.h | grep CUDNN_MAJOR -A 2

在这里插入图片描述

下载TensorRT8.5 GA

下载地址：https://developer.nvidia.com/nvidia-tensorrt-8x-download

在这里插入图片描述
进行解压
在这里插入图片描述

配置环境变量

# 配置环境变量
sudo gedit ~/.bashrc #复制到最后面


export PATH=$PATH:/TensorRT/TensorRT-8.5.1.7/bin
export LD_LIBRARY_PATH=/TensorRT/TensorRT-8.5.1.7/lib:$LD_LIBRARY_PATH


# 保存后刷新环境变量
source ~/.bashrc

查看是否配置成功，我配置的是8.5.1版本

trtexec

在这里插入图片描述

进行编译

进入这个文件目录下
cd samples/sampleOnnxMNIST


进行编译命令
sudo make -j16

报错：NvInferRuntimeCommon.h:19:10: fatal error: cuda_runtime_api.h: 没有那个文件或目录等

在这里插入图片描述

查看有没有cuda的链接，没有看到cuda的链接就是没有配置cuda软连接

cd /usr/local/
ls -l

配置软连接

sudo ln -s /cuda/cuda118 /usr/local/cuda

在进行编译

cd  TensorRT/TensorRT-8.5.1.7/samples/sampleOnnxMNIST

sudo make -j16

查看是否成功

cd TensorRT/TensorRT-8.5.1.7/bin

sample_onnx_mnist

在这里插入图片描述


http://www.kler.cn/a/132298.html

相关文章：

  • mysql8.4+mysql router读写分离
  • 【网络】Socket编程TCP/UDP序列化和反序列化理解应用层（C++实现）Json::Value
  • 小鹏汽车嵌入式面试题及参考答案
  • IDEA2023 SpringBoot整合Web开发（二）
  • Ascend C算子性能优化实用技巧05——API使用优化
  • 深入理解Redis（七）----Redis实现分布式锁
  • Springboot2+thymeleaf+Jpa实现CRUD操作
  • 安全框架springSecurity+Jwt+Vue-1(vue环境搭建、动态路由、动态标签页)
  • 一文了解VR全景拍摄设备如何选择，全景图片如何处理
  • 关于SPJ表的数据库作业
  • frp新版本frp_0.52.3设置
  • 【SA8295P 源码分析】127 - 摄像头 GMSL1、GMSL2 加串-解串器 常用寄存器配置整理（正在更新中......）
  • 【项目管理】PMO技能树21项参照
  • duplicate复制数据库单个数据文件复制失败报错rman-03009 ora-03113
  • 数据库Communications link failure
  • 记录一次macos没有sudoers文件问题
  • PSP - 蛋白质复合物结构预测 Template 的 Multichain Mask 2D (二维多链掩码)
  • MFC保存窗口客户区为图片
  • BGP的基础知识
  • 人工智能Keras的第一个图像分类器（CNN卷积神经网络的图片识别）
  • 【备忘】ChromeDriver 官方下载地址 Selenium，pyppetter依赖
  • 直播间停留人气的10种方法
  • JAVA刷题之字符串的一些个人思路
  • C# 正则表达式大全
  • scala解析命令行参数
  • 将 ONLYOFFICE 文档编辑器与 Node.js 应用集成