1.使用pip、conda能够直接安装上最好。
pip install thop
conda install thop
2.本地安装
(1)去官网下载压缩包
pytorch-OpCounter:基于 PyTorch 的算子计数工具项目 - AtomGit
(2)先激活虚拟环境,再手动一步步cd到指定文件夹下
eg: ..\pytorch\Lib\site-packages
(3)尝试运行setup.py
如果出现Finished processing ...就代表安装成功了
(4)额外安装失败解决方式
本人是遇到了上述步骤进行完也无法安装的问题。
error:packaging.version.InvalidVersion: Invalid version: '0.1.1_2607282120'
原因是thop 部分源码仓库里的setup.py自动生成版本号用了下划线0.1.1_2607282120,新版packaging严格遵循 PEP440,直接抛异常。 本地拉源码手动 python setup.py install 时报错。
修改错误也很简单,在setup.py文件中找到这一句:
VERSION += "_" + datetime.datetime.now ().strftime ("% Y% m% d% H% M")[2:]
然后将非法字符“_”修改为“.dev”,如下:
VERSION += ".dev" + datetime.datetime.now ().strftime ("% Y% m% d% H% M")[2:]
再回到命令窗口运行修改完毕的setup.py文件就可以了
python setup.py install