摘要: 代码流程如下: 1.加在权重 2.创建builder 3.利用builder创建网络定义network以及配置文件config 4.利用builder、network、config创建网络 包括:a)搭建网络结构,b)创建引擎 5.推理 a)创建处理缓存区与引擎交互的内存buffer b)利用引擎创 阅读全文
posted @ 2020-11-09 21:49 快乐码小农 阅读(221) 评论(0) 推荐(0) 编辑
摘要: 本人工作环境为: windows7 vs2015 cuda10.2 cudnn版本请用7.6.5,如果是8.0,运行的时候会报错:计算机中丢失cudnn_7.dll 配置TensorRT流程如下: 1.首先下载TensorRT7.0 https://developer.nvidia.com/nvid 阅读全文
posted @ 2020-11-09 20:07 快乐码小农 阅读(1377) 评论(0) 推荐(0) 编辑