网站介绍：文章浏览阅读135次。一、验证集评估阶段的张量计算非常占用空间。法1. 可以对利用torch.tensor().detach().cpu().numpy()转为numpy，在cpu上进行loss和acc的计算法2. 直接对评估阶段使用with torch.no_grad():_torch.tensor.cpu().detach() torch.cuda.outofmemoryerror: cuda out of memory
- 链接地址：https://blog.csdn.net/a18379692263/article/details/120745340
- 链接标题：pytroch运行时出现：CUDA error:out of memory_torch.tensor.cpu().detach() torch.cuda.outofmemory-CSDN博客
- 所属网站：blog.csdn.net
- 被收藏次数：3199
- 网站标签：torch.tensor.cpu().detach() torch.cuda.outofmemoryerror: cuda out of memory
