TensorRT软件包onnx转TensorRT运行的例程,高版本不兼容,源代码如下
#include
#include <assert.h>
#include
#include <cuda_runtime_api.h>
#include
#include
#include
#include
#include <sys/stat.h>
#include <time.h>
#include “NvInfer.h”
#include “NvOnnxParser.h”
#include “common.h”
using namespace
原创
2021-08-29 19:50:59 ·
830 阅读 ·
0 评论