五分钟搞定VS2017+TensorRT环境搭建
点击上方“小白学视觉”,选择加"星标"或“置顶”
重磅干货,第一时间送达
本文转自|OpenCV学堂
系统与环境要求
Windows10系统
GTX1050Ti
CUDA10.x
VS2017
TensorRT7.0.0.11
01 安装与配置
下载路径:
https://developer.nvidia.com/TensorRT
首先需要下载TensorRT的ZIP格式文件到本地,然后解压缩到
D:\TensorRT-7.0.0.11
然后打开VS2017,新建一个空项目,分别配置
1. 包含目录
D:\TensorRT-7.0.0.11\include
2. 库目录
D:\TensorRT-7.0.0.11\lib
3. 链接器
myelin64_1.lib
nvinfer.lib
nvinfer_plugin.lib
nvonnxparser.lib
nvparsers.lib
4. 环境变量
D:\TensorRT-7.0.0.11\lib
然后在系统的环境变量中添加:
重启VS即可。
02 代码验证与测试
2020年初,我写过的pytorch程序有个Hello Wrold的版本的模型就是mnist.onnx,我来测试一下是否可以通过TensorRT来实现对ONNX格式模型加载。重启VS2017之后在原来的空项目上然后添加一个cpp文件,把下面的代码copy到cpp文件中:
#include <fstream>
#include <iostream>
#include <sstream>#include "NvInfer.h"
#include "NvOnnxParser.h"using namespace nvinfer1;
using namespace nvonnxparser;class Logger : public ILogger
{void log(Severity severity, const char* msg) override{// suppress info-level messagesif (severity != Severity::kINFO)std::cout << msg << std::endl;}
} gLogger;int main(int argc, char** argv) {std::string onnx_filename = "D:/python/pytorch_tutorial/cnn_mnist.onnx";IBuilder* builder = createInferBuilder(gLogger);nvinfer1::INetworkDefinition* network = builder->createNetworkV2(1U << static_cast<uint32_t>(NetworkDefinitionCreationFlag::kEXPLICIT_BATCH));auto parser = nvonnxparser::createParser(*network, gLogger);parser->parseFromFile(onnx_filename.c_str(), 2);for (int i = 0; i < parser->getNbErrors(); ++i){std::cout << parser->getError(i)->desc() << std::endl;}printf("tensorRT load onnx mnist model...\n");return 0;
}
编译运行直接运行输出:
恭喜你!TensorRT在Windows10下开发环境配置成功了!绝对在5分钟内搞定,前提是先预装好前面说的那些依赖软件与相关的库!
还不清楚,看B站视频:
https://www.bilibili.com/video/BV1Bf4y167Ty
好消息,小白学视觉团队的知识星球开通啦,为了感谢大家的支持与厚爱,团队决定将价值149元的知识星球现时免费加入。各位小伙伴们要抓住机会哦!
下载1:OpenCV-Contrib扩展模块中文版教程
在「小白学视觉」公众号后台回复:扩展模块中文教程,即可下载全网第一份OpenCV扩展模块教程中文版,涵盖扩展模块安装、SFM算法、立体视觉、目标跟踪、生物视觉、超分辨率处理等二十多章内容。
下载2:Python视觉实战项目52讲
在「小白学视觉」公众号后台回复:Python视觉实战项目,即可下载包括图像分割、口罩检测、车道线检测、车辆计数、添加眼线、车牌识别、字符识别、情绪检测、文本内容提取、面部识别等31个视觉实战项目,助力快速学校计算机视觉。
下载3:OpenCV实战项目20讲
在「小白学视觉」公众号后台回复:OpenCV实战项目20讲,即可下载含有20个基于OpenCV实现20个实战项目,实现OpenCV学习进阶。
交流群
欢迎加入公众号读者群一起和同行交流,目前有SLAM、三维视觉、传感器、自动驾驶、计算摄影、检测、分割、识别、医学影像、GAN、算法竞赛等微信群(以后会逐渐细分),请扫描下面微信号加群,备注:”昵称+学校/公司+研究方向“,例如:”张三 + 上海交大 + 视觉SLAM“。请按照格式备注,否则不予通过。添加成功后会根据研究方向邀请进入相关微信群。请勿在群内发送广告,否则会请出群,谢谢理解~
五分钟搞定VS2017+TensorRT环境搭建相关推荐
- 启程 - 《每日五分钟搞定大数据》
<每日五分钟搞定大数据>原创系列,每周不定期更新.欢迎关注公众号:大叔据 想了很久,准备开始写一系列的文章,记录这些年来的所得所想,感觉内容比较多不知从哪里开始,画了个思维导图确定了大 ...
- 正则表达式里转义字符_五分钟搞定正则表达式,如果没搞定,再加两分钟
五分钟搞定正则表达式,如果没搞定,再加两分钟 [这是 ZY 第 18 篇原创文章] 文章概览 一.正则表达式介绍 正则表达式,又称规则表达式.(英语:Regular Expression,在代码中常简 ...
- 五分钟搞定5000字-外文文献翻译,…
五分钟搞定5000字-外文文献翻译 工具大全 http://www.icoolen.com/node/2151 建议收藏 在科研过程中阅读翻译外文文献是一个非常重要的环节,许多领域高水平的文献都是外文 ...
- 用 Docker 搞定 PHP 开发环境搭建
避坑!用 Docker 搞定 PHP 开发环境搭建(Mac.Docker.Nginx.PHP-FPM.XDebug.PHPStorm.VSCode) 因为最近换电脑的缘故需要重新部署一下本地的开发环境 ...
- 五步搞定Android开发环境部署——非常详细的Android开发环境搭建教程(转)
引言 在windows安装Android的开发环境不简单也说不上算复杂,本文写给第一次想在自己Windows上建立Android开发环境投入Android浪潮的朋友们,为了确保大家能顺利完成开发环境的 ...
- 五分钟搞定:Centos中Kafka和Zookeeper的快速安装教程
[提前声明] 文章由作者:张耀峰 结合自己生产中的使用经验整理,最终形成简单易懂的文章 写作不易,转载请注明,谢谢! 代码案例地址: ?https://github.com/Mydreamandrea ...
- 五分钟搞定 HTTPS 配置,二哥手把手教
01.关于 FreeSSL.cn FreeSSL.cn 是一个免费提供 HTTPS 证书申请.HTTPS 证书管理和 HTTPS 证书到期提醒服务的网站,旨在推进 HTTPS 证书的普及与应用,简化证 ...
- 吹牛的资本之Hibernate框架,五分钟搞定Hibernate...
Hibernate框架: (ps:有标记"百度网盘可看"的地方可在网盘中下载配有注释的原代码) 现在努力的敲代码,是为了以后可以选择敲不敲代码! 百度网盘:http://pan.b ...
- HyperLPR3车牌识别-五分钟搞定: 中文车牌识别光速部署与使用
简介 HyperLPR在2023年初已经更新到了v3的版本,该版本与先前的版本一样都是用于识别中文车牌的开源图像算法项目,最新的版本的源码可从github中提取:https://github.com/ ...
最新文章
- 深度学习--TensorFlow(6)神经网络 -- 拟合线性函数非线性函数
- Java(三):序列化
- 2021春季每日一题 【week2 未完结】
- linux下查看cmake的版本
- SpringMVC 日期类型转换
- js操作select标签
- mysqlinsert触发器的创建
- c语言归并排序数组不固定,归并排序非递归实现C语言
- 001. 为input type=text 时设置默认值
- read()/write()的生命旅程之三——第三章:write()
- 究竟云计算就业前景好不好 零基础如何学云计算
- 测试开发面试题160道17类21339字
- Ubuntu16.04.4制作自定义安装光盘方案
- Linux系统 查看 Vendor id 和Device id
- 怎样用sc命令运行服务器,windows中的sc控制服务命令
- RPG游戏Demo学习笔记一
- Python安装读取Excel的xlrd模块步骤
- I3D泛读【Que Vadis,Action Recognition?A New Model and the Kinetics Dataset】
- 史上最简单的rar压缩包文档解密方法,rar压缩包权限密码如何解开?
- ic和mos怎么区分_MOS管和IGBT管的定义是什么与怎么辨别呢?