banxia1995 2019-04-12 10:16 采纳率: 0%
浏览 5393

linux下使用pytorch框架出现cuda run out of memory问题

1.使用pytorch，数据量是图像224x224，总共4w张，框架使用的是VGG，出现cuda memory问题
图片说明

上图是gpu使用的情况，运行时使用的batch_size为32

显存问题怎么解决呢？

求大神指点指点，实在不知道怎么解决了。。。

  • 写回答

1条回答 默认 最新

  • 吃鸡王者 2019-04-12 16:47
    关注

    检查一下是不是一次性把所有的图片都加载进去了

    评论

报告相同问题？