weixin_46908570 2020-12-09 14:58 采纳率: 0%
浏览 834

运行yolov4的时候报错CUDA Error: out of memory是什么原因

 

  • 写回答

1条回答 默认 最新

  • 一颗小树x 领域专家: 人工智能技术领域 2020-12-12 09:41
    关注

    你好,这里是说CUDA内存不足。先用nvidia-smi命令来查看某一时刻的GPU的占用情况,GPU是否真的都被使用了。如果不是,看看这篇文章https://www.cnblogs.com/jisongxie/p/10276742.html

    设置CUDA_VISIBLE_DEVICES

    评论

报告相同问题?