YOLOv7报错，不知道应该用哪个torch版本

运行报错
AttributeError: module 'torch' has no attribute 'concat'

下面是torch版本：

torch: 1.7.0+cu110
torchvision: 0.8.0
GPU: True
1

应该怎么解决这个报错

这个错误表明在你的代码中使用了 torch.concat 这个函数，但是在你的 torch 版本中，这个函数并不存在。这可能是因为你的代码需要更高版本的 torch。

在 PyTorch 1.7.0 版本中，torch.concat 已经被重命名为 torch.cat。所以你需要将代码中的所有 torch.concat 替换为 torch.cat。

例如，如果你有以下代码：
output = torch.concat([output1, output2], dim=1)
你应该修改为：
output = torch.cat([output1, output2], dim=1)
在解决这个问题之后，你可能需要升级你的 torch 和 torchvision 到最新版本，因为新版本会有更多的功能和修复。你可以使用 pip 来升级这些库：
pip install --upgrade torch torchvision
这将安装最新版本的 torch 和 torchvision。

【以下回答由 GPT 生成】

由于你的torch版本为1.7.0+cu110，而出现的报错是AttributeError: module 'torch' has no attribute 'concat'，这意味着在torch v1.7.0中concat函数已被弃用。

要解决这个问题，你可以升级torch版本到1.9.0以上，或者使用其他与YOLOv7兼容的torch版本。以下是两种解决方案:

解决方案一: 升级torch版本

步骤1: 更新torch和torchvision版本。

pip install torch torchvision --upgrade

步骤2: 确认torch和torchvision版本已成功升级。

import torch
import torchvision

print(torch.__version__)
print(torchvision.__version__)

如果版本号显示为1.9.0或更高，则表示升级成功。

解决方案二: 选择与YOLOv7兼容的torch版本

如果你不想升级torch版本，可以尝试使用其他与YOLOv7兼容的torch版本。以下是一些可能与YOLOv7兼容的torch版本: - torch==1.6.0 - torch==1.8.0 - torch==1.9.0

步骤1: 安装特定版本的torch。

pip install torch==1.6.0

步骤2: 确认torch版本已成功安装。

import torch

print(torch.__version__)

如果版本号显示为你选择的torch版本号，则表示安装成功。

请注意，根据你的CUDA版本，你可能需要选择与相应CUDA版本兼容的torch版本。如果存在其他报错，请参考相关错误提示进行修复。

希望以上解决方案能够帮助你解决问题。如果你还有其他问题，请随时提问。