pytorch gpu版 yolov5 报错

错误一：

AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'

 解决方法，修改pytorch 源码：

Lib\site-packages\torch\nn\modules下找到文件upsampling.py

    def forward(self, input: Tensor) -> Tensor:
        # return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners,
        #                      recompute_scale_factor=self.recompute_scale_factor)
        return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)

错误二：

RuntimeError: result type Float can't be cast to the desired output type xxx

yolov5下的 【utils 文件夹】中的【loss.py】

需要修改的两处（注释的代码时之前的代码，下面的是修改的）：

 修改的两处类容如下：

anchors, shape = self.anchors[i], p[i].shape  # 修改，2022年7月11日
​​​​​​​
indices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1)))  # image, anchor, grid  # 修改，2022年7月11日