安装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")

 

你可能感兴趣的:(深度学习)