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

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

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

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

显存问题怎么解决呢？

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

  • 写回答

1条回答 默认 最新

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

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

    评论

报告相同问题？

悬赏问题

  • ¥15 请教一下各位，为什么我这个没有实现模拟点击
  • ¥15 执行 virtuoso 命令后，界面没有，cadence 启动不起来
  • ¥50 comfyui下连接animatediff节点生成视频质量非常差的原因
  • ¥20 有关区间dp的问题求解
  • ¥15 多电路系统共用电源的串扰问题
  • ¥15 slam rangenet++配置
  • ¥15 有没有研究水声通信方面的帮我改俩matlab代码
  • ¥15 ubuntu子系统密码忘记
  • ¥15 保护模式-系统加载-段寄存器
  • ¥15 电脑桌面设定一个区域禁止鼠标操作