tony365的博客

比如模型infer输出是c,h,w, float32的结果，后处理 input是 c,h,w float32 的数据，output转换为 h,w,c float32的数据（对应上面的cv::Mat result(img_h, img_w, CV_32FC3);然后前处理，infer, 后处理 内部也有自己的input和output，不要搞混淆了。前处理，infer , 后处理是一个 graph , 也就是demo中完整的图。demo中的input和output是 完整的图的输入输出。

上面的教程是通过https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html Package Manager Installation的方法安装cudnn的。sudo apt-get install cuda-drivers-418 安装特定版本的驱动。ubuntu-drivers devices 寻找可以安装的驱动。除了以上的 通过下runfile来安装，也可以通过包安装。Ctrl + Alt + Delete 退出tty模式。