from thop import clever_format, profile

 出现以下报错:

在 Anaconda Prompt (anaconda3) 中输入以下命令:

# 先激活你的虚拟环境
activate pytorch

# 然后在虚拟环境下安装
pip install thop -i https://pypi.tuna.tsinghua.edu.cn/simple somepackages

 就好了 ~


看了挺多篇文章,试了三四种都会报错,觉得有必要记录一下。

安装参考的文章:

Win10下Anaconda安装功能包pycocotools+opencv-python

关于thop库的使用:

采用thop库正确计算模型计算量FLOPs和模型参数Params 

一起学习

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐