安装thop计算网络FLOPs

thop安装

配置yolov5环境的时候,thop可能会遇到安装错误,直接pip会失败。
以下链接是thop包github链接:
https://github.com/Lyken17/pytorch-OpCounter

可行方法(安装成功):
第一种:

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

第二种:(推荐)
1、找到thop包的github链接(https://github.com/Lyken17/pytorch-OpCounter),下载压缩文件,解压
2、打开Anaconda Prompt,激活自己的pytorch环境,cd到解压文件路径
3、输入命令 python setup.py install,等待安装即可

猜你喜欢

转载自blog.csdn.net/retainenergy/article/details/124519317