在conda虚拟环境中安装thop:
在ubuntu16.04下亲测成功,首先要进入到相应的虚拟环境,
例如虚拟环境为my_env

source activate my_env

然后再接着输入下面的命令。

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

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

更多推荐