工程化模型:使用TensorRT-在Linux下的加速推理,从环境安装到训练部署全套流程
文章目录1、环境与版本说明2、环境安装2.1、CUDA 安装驱动drivers切换gcc安装cuda安装cudnn2.2、安装TensorRT2.3、安装TF与其他环境3、模型训练(以InceptionV3为例)-->略4、使用tensorRT推理加速4.1、模型保存4.2、生成PB文件4.3、pb转换成uff准备工作参数设置pb转uff 并解析模型保存模型4.4、推理步骤4.5、完整代码1、环境与版本说明 ~~~~
原创
2021-06-08 15:59:12 ·
1018 阅读 ·
2 评论