TensorRT：NvInfer.h-上篇

版本：5.1.5.0

工欲善其事必先利其器，后面需要使用RT插件，所以还是得学习一下RT的相关代码

NvInfer.h中声明和定义了RT中的大部分变量、函数、类等

RT中代码格式较为统一，大体看了设定规则

枚举变量：k+全大写， kFLOAT， kGPU，kCHANNEL类内变量：k+首字母大写，nbDims类内函数：小写+首字母大写，setNames(), setInput()不能派生或者继承的：用大写的“I”进行了标识， ITensor,ILayer

具体的接口：https://docs.nvidia.com/deeplearning/tensorrt/api/c_api 可以在这里查看

0：命名空间：nvidia + infer --- > nvinfer1

namespace nvinfer1

1：枚举变量

数据类型：float32, float16, int8, int32


enum class DataType : int
{kFLOAT = 0, //!< FP32 format.kHALF = 1,  //!< FP16 format.kINT8 =

TensorRT：NvInfer.h-上篇相关推荐

NvInfer.h: No such file or directory
编译tensorrt yolov5版本的时候遇到这个错误. 我是默认你安装了tensorRT之后还是遇到这个错误,如果没有安装遇到这个错误,请安装之后再来看,如果已经安装了,那么就修改CMakeLis ...
TensorRT升级错误,plugins/poly_layer.h:19:21: error: looser throw specifier for ‘virtual int nvinfer1::
在将tensorrt由7升级到8的过程中,遇到了如下错误 In file included from plugins/poly_layer.cpp:1:0: plugins/poly_layer.h: ...
AArch64编译安装特定GPU版本TensorFlow及Bazel
前排提示:如果使用的 cuDNN 版本高于 7,会无法编译安装带 CUDA 支持的 TensorFlow 1 版本. 本文基于 Jetson TX2. ALBERT 的 requirements.tx ...
【c++】——通过cmake定义宏add_definitions
亮点:通过cmake定义的选项在代码中进行动态选择源码:dler 1. 例子代码中通过宏 tx2 进行代码的控制． code #ifndef NET_H #define NET_H#include ...
TensorRT简介
TensorRT 介绍引用:https://arleyzhang.github.io/articles/7f4b25ce/ 1 简介 TensorRT是一个高性能的深度学习推理(Inference) ...
TensorRT Samples: MNIST(serialize TensorRT model)
关于TensorRT的介绍可以参考: http://blog.csdn.net/fengbingchun/article/details/78469551 这里实现在构建阶段将TensorRT mod ...
TensorRT Samples: MNIST(Plugin, add a custom layer)
关于TensorRT的介绍可以参考:http://blog.csdn.net/fengbingchun/article/details/78469551 以下是参考TensorRT 2.1.2中的sa ...
TensorRT Samples: CharRNN
关于TensorRT的介绍可以参考: http://blog.csdn.net/fengbingchun/article/details/78469551 以下是参考TensorRT 2.1.2中的 ...
TensorRT Samples: GoogleNet
关于TensorRT的介绍可以参考: http://blog.csdn.net/fengbingchun/article/details/78469551 以下是参考TensorRT 2.1.2中的 ...

TensorRT：NvInfer.h-上篇

TensorRT：NvInfer.h-上篇相关推荐

最新文章

热门文章