admin管理员组

文章数量:1611566

跑yolov4测试demo报错

RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 1.96 GiB total capacity; 1.40 GiB already allocated; 18.50 MiB free; 1.46 GiB reserved in total by PyTorch)

原因

  1. GPU还有其他进程占用显存，导致本进程无法分配到足够的显存
  2. 缓存过多，使用torch.cuda.empty_cache()清理缓存
  3. 显存不够，换块显卡吧

解决方案

测试的时候加上

with torch.no_grad():
  # test process

本文标签： allocateMIBRuntimeErrorCUDAMemory

版权声明：本文标题：RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0； 1.96 GiB total capacity； 1.40 内容由热心网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：https://m.elefans.com/dianzi/1728620873a1166343.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。