pytorch出现CUDA error:out of memory错误
pytorch出现CUDA error:out of memory错误问题描述解决方案问题描述模型训练过程中报错,提示CUDA error:out of memory。解决方案判断模型是否规模太大或者batchsize太大,可以优化模型或者减小batchsize;比如:已分配的显存接近主GPU的总量,且仍需要分配的显存大于缓存(306M>148.3M)。cached(...
原创
2020-04-10 23:51:13 ·
5110 阅读 ·
0 评论