pytorch
lyf_666
波澜不惊
展开
-
RuntimeError: CUDA out of memory. Tried to allocate 模型训练 GPU 显存不够报错总结
RuntimeError: CUDA out of memory. Tried to allocate 模型训练 GPU 显存不够报错总结原创 2022-08-07 00:18:14 · 833 阅读 · 1 评论 -
pytorch训练神经网络爆内存解决办法
pytorch训练神经网络爆内存解决办法训练的时候内存一直在增加,最后内存爆满,被迫中断。后来换了一个电脑发现还是这样,考虑是代码的问题。检查才发现我的代码两次存了loss,只有一个地方写的是loss.item()。问题就在loss,因为loss是variable类型。要写成loss_train = loss_train + loss.item(),不能直接写loss_train = loss_train + loss。否则就会发现随着epoch的增加,占的内存也在一点一点增加。算是一个小坑吧,原创 2021-03-23 11:09:59 · 4380 阅读 · 0 评论 -
tensor转成numpy的几种情况
下面将将tensor转成numpy的几种情况GPU中的Variable变量:a.cuda().data.cpu().numpy()GPU中的tensor变量:a.cuda().cpu().numpy()CPU中的Variable变量:a.data.numpy()CPU中的tensor变量:a.numpy()总结:.cuda()是读取GPU中的数据.data是读取Variable中的tensor.cpu是把数据转移到cpu上.numpy()把tensor变成num原创 2020-08-21 15:56:56 · 885 阅读 · 0 评论