1.CUDA内存不足
我在使用yolov7中遇到了一些问题，通过查询相关的资料和网站将其一个一个解决了。首先遇到的第一个问题就是cuda的内存不足，报错内容如下所示：
RuntimeError: CUDA out of memory. Tried to allocate 52.00 MiB (GPU 0; 5.80 GiB total capacity; 4.62 GiB already allocated; 36.38 MiB free; 4.64 GiB reserved in total by PyTorch) If reserved memory is通过分析它主要有以下三种原因。
应该有三个原因
-
GPU还有其他进程占用显存，导致本进程无法分配到足够的显存
-
缓存过多，使用torch.cuda.empty_cache()清理缓存
-
卡不行，换块显存更大的卡
解决办法
1.最直接的解决办法是减小batch_size;
python train_net.py --config-file configs/coco/darknet53.yaml --num-gpus 1如果使用以上命令，可以在 configs/coco/darknet53.yaml文件下使用ctrl+f搜索batch 将其修改为16，还是报同样错误则修改为8。
2.如果在训练过程遇到这种情况,可以尝试在训练前先释放CUDA内存 nvidia-smi查看GPU使用率,如果使用率不高,就使用torch.cuda.empty_cache()释放内存。
在train_net.py中main下面添加如下代码：
if hasattr(torch.cuda, 'empty_cache'):
torch.cuda.empty_cache()以上方法即可解决。
2.UserWarning
报错内容如下：
/yolov7-1.0/yolov7/modeling/meta_arch/yolov7.py:704: UserWarning: __floordiv__ is deprecated, and its behavior will change in a future version of pytorch. It currently rounds toward 0 (like the 'trunc' function NOT 'floor'). This results in incorrect rounding for negative values. To keep the current behavior, use torch.div(a, b, rounding_mode='trunc'), or for actual floor division, use torch.div(a, b, rounding_mode='floor').
best_n_mask = (best_n_all // 3) == self.level
报错截图如下所示：
按照要求找到yolov7-1.0/yolov7/modeling/meta_arch/yolov7.py文件的704行
将其修改为如下内容：文章来源：https://www.toymoban.com/news/detail-473713.html
best_n_mask=torch.div(best_n_all, 3, rounding_mode='floor')== self.level再次运行下面命令即可成功。文章来源地址https://www.toymoban.com/news/detail-473713.html
python train_net.py --config-file configs/coco/darknet53.yaml --num-gpus 1到了这里，关于使用Yolov7遇到的一些问题的解决方法的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
