【笔记】安装pytorch thop模块的BUG解决

如果大家的pip install没有问题，直接使用pip install thop即可。

但是博主使用pip 安装得时候出现问题。我的user权限并不是和root一样，但是pip3 install 直接就安装到了/usr的目录下，这导致我的conda虚拟环境无法读取其中的库。个人猜测可能是软连接的时候路径有错误，应该软连接自己的user下的pip3 lib。

但由于急着安装好thop来看自己模型的参数

所以博主试了很多方法，最后自己急中生智想到可不可以直接安装源码呢，结果1分钟不到问题解决。

源码地址：https://github.com/Lyken17/pytorch-OpCounter

下载下来之后 python setup.py install 立马就安装好可以使用了。

在此顺便记录下thop看model的flops和parameters的语句。

flops, params = profile(net, inputs =(input))
flops, params = clever_format([flops, params], "%.3f")