问题描述:在跑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。
解决办法:我尝试把电脑重启了一下,然后问题就解决了。
之前遇到这个错误,网上看见的说法都是说电脑显存不够用,估计换个更高配置的显卡也能解决这个问题。
其实,我电脑配置不低了,如下图所示,所以也可能是其他原因导致的,但是除了重启电脑解决,其他办法我还不知道。