[已解決]No module named ‘torchsummary‘

報錯如下

原因分析

torchsummary

可以做Pytorch可視化，輸出網絡相關資訊。目前環境缺失

torchsummary

安裝包。

解決方案

pip install torchsummary

注：conda install torchsummary可能無法安裝，那就直接用pip吧。

搞定！