admin管理员组

文章数量:1794759

python中对thop安装问题

python中对thop安装问题

背景：自己网络模型的结果不是特别好，然后就想多路径出发。从模型参数和处理速度上去与其他模型进行对比，尝试去找到我的模型存在的必要性。故：接触了thop这个第三方安装库，狠狠的踩了一个雷。今天记录一下，也避免大家重复踩雷。

thop的安装问题

在python中安装过第三方库的小伙伴都知道，它的安装命令是：pip install XXX。然而针对Thop的安装却有一个大坑。

1）直接使用常用命令

pip install thop

它会默认安装0.0.31.post2005241907的版本。由于我使用的pytorch环境为1.0.0，不支持： nn.SyncBatchNorm（多GPU执行所需的），而该版本下的THop【THOP:0.031】没有对pytorch:1.0.0修复此bug，从而导致运行出错。

2）使用git库安装方式

pip install --upgrade git+github/Lyken17/pytorch-OpCounter.git

第二种方式安装的就会是最新的Thop库的结果！！！超级方便！！！

本文标签： 中对Pythonthop

版权声明：本文标题：python中对thop安装问题 内容由林淑君副主任自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.xiehuijuan.com/baike/1686909230a117077.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。