YOLOv7-pose运行过程中出现RuntimeError: CUDA out of memory. Tried to allocate 8.14 GiB的错误。_yolov7训练: cuda out of memory. tried to allocate 8.

作者：IT小白 | 2024-03-11 20:56:26

踩

yolov7训练: cuda out of memory. tried to allocate 8.57 gib (gpu 0; 24.00 gib

问题描述：在跑YOLOv7-pose代码时，出现了下面这个错误RuntimeError: CUDA out of memory. Tried to allocate 8.14 GiB (GPU 0; 12.00 GiB total capacity; 1.65 GiB already allocated; 8.15 GiB free; 1.85 GiB reserved in total by PyTorch)。事实上，之前跑YOLOv5代码的时候也出现了这个报错，具体YOLOv5的哪个版本就忘了。怀疑是参数量大的版本，因为YOLOv7-pose的权重文件是161.2MB。

解决办法：我尝试把电脑重启了一下，然后问题就解决了。

之前遇到这个错误，网上看见的说法都是说电脑显存不够用，估计换个更高配置的显卡也能解决这个问题。

其实，我电脑配置不低了，如下图所示，所以也可能是其他原因导致的，但是除了重启电脑解决，其他办法我还不知道。

本文内容由网友自发贡献，转载请注明出处：https://www.wpsshop.cn/w/IT小白/article/detail/220596