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