python安装thop

要在 Python 中使用 THOP，需要先安装 PyTorch。然后可以使用以下命令在命令行中安装 THOP：

pip install thop

确保你的 pip 版本是最新的，可以使用以下命令升级 pip：

pip install --upgrade pip

安装完成后，你可以在 Python 中导入 THOP，例如：

import torch
from thop import profile, clever_format

model = torch.nn.Sequential(
          torch.nn.Conv2d(3, 32, kernel_size=3, stride=1, padding=1),
          torch.nn.ReLU(),
          torch.nn.Conv2d(32, 64, kernel_size=3, stride=1, padding=1),
          torch.nn.ReLU(),
          torch.nn.MaxPool2d(kernel_size=2, stride=2),
          torch.nn.Conv2d(64, 128, kernel_size=3, stride=1, padding=1),
          torch.nn.ReLU(),
          torch.nn.Conv2d(128, 256, kernel_size=3, stride=1, padding=1),
          torch.nn.ReLU(),
          torch.nn.MaxPool2d(kernel_size=2, stride=2),
          torch.nn.Conv2d(256, 512, kernel_size=3, stride=1, padding=1),
          torch.nn.ReLU(),
          torch.nn.Conv2d(512, 1024, kernel_size=3, stride=1, padding=1),
          torch.nn.ReLU(),
          torch.nn.MaxPool2d(kernel_size=2, stride=2),
          torch.nn.Flatten(),
          torch.nn.Linear(1024 * 4 * 4, 10)
        )

input = torch.randn(1, 3, 32, 32)
flops, params = profile(model, inputs=(input,))
flops, params = clever_format([flops, params], "%.3f")
print(f"FLOPs: {flops}, Params: {params}")

这个例子演示了如何使用 THOP 来计算一个简单的 CNN 模型的 FLOPs 和参数数量。你可以根据自己的需求修改这个例子。