YOLO11报错：AttributeError: module ‘torch‘ has no attribute ‘OutOfMemoryError‘

事情是这样的：前几天YOLO11的代码还是可以训练的，昨天训练了一天，今天换模型就报这个错。

AttributeError: module 'torch' has no attribute 'OutOfMemoryError'

我查了一下：YOLO11官方代码issues里面也有人有同样的问题，不过给出的解决方法是：

检查数据集中的所有标签是否有错误

我的情况是前几天能用，今天突然不能用，显然不是这个问题。

链接在这：https://github.com/ultralytics/ultralytics/issues/18433

后面我又看到一个回答，说是YOLO11代码中有BUG，但是作者回答说2023年1月已经修复了漏洞。所以如果你用的YOLO11比较早，建议换最新的代码

链接在这：https://github.com/ultralytics/ultralytics/issues/487

后来我细看了一下，产生BUG的原因是Pytorch的版本问题，说上面的报错只有高一点的Pytorch才有，于是我升级到最新的Pytorch，问题就解决了。

查资料看有人说降低Batch-size也能行，我没试，如果上面的思路都解决不了，就试试吧~