搭建流程 模型转换及预测 通用函数编写 定义一个G数据占用内存量 def GiB(val): return val * 1 << 30 构建engine函数 (pytorch逐层解析参考链接: https://gitee.com/xn1997/TensorRT-sample/