网上查阅了一些资料,包括什么改虚拟内存,都没什么用,最后我修改了batch_size,把它改小,模型终于能跑了
pytorch模型提示超出内存cuda runtime error(2): out of memory
猜你喜欢
转载自blog.csdn.net/weixin_43570254/article/details/104157532
今日推荐
周排行