爪爪熊的博客

想使用libtorch建一个推理服务器，前端web准备使用rust+axum，后端推理准备尝试c++来做：前言：我所有的操作都使用vmware中的虚拟机ubuntu 20.04完成。有如下几个方便的点：性能没啥问题可以建立快照保存，如果怀疑下面的操作有问题，可以提前建立快照出问题了可以及时还原回来。主力机是amd win10笔记本，直接装ubuntu双系统，显卡驱动不正常，亮度不能调节等毛病。​正文：一、libtorch&opencv4&grpc编译1.1 grpc编译

爪爪熊原创文章----谢绝转载本文解决问题：torchsummary针对多个输入模型的时候，其输出信息中input size等存在着错误，这里提供方案解决这个错误。当我们使用pytorch搭建好我们自己的深度学习模型的的时候，我们总想看看具体的网络信息以及参数量大小，这时候就要请出我们的神器 torchsummary了，torchsummary的简单使用如下所示：# pip install torchsummaryfrom torchsummary import summarymodel