如何解决此问题“ RuntimeError：CUDA内存不足。”？

Question

我将从图片中提取特征。首先定义张量data_feature_map，然后使用torch.cat堆叠一张图片的特征。

我的代码是：

data_feature_map = torch.ones(1,2048)
for i, data in enumerate(train_loader, 0):
    img, _ = data
    img.requires_grad_=False
    if torch.cuda.is_available():
        img = img.cuda()
    out = model(img)
    # out.shape = [1,2048]
    out = out.view(1,-1).cpu()
    data_feature_map = torch.cat((data_feature_map, out), 0)

但是当我运行它时，它将显示错误"RuntimeError: CUDA out of memory."

请告诉我为什么会发生此错误。非常感谢。

Answer 1

由于GPU的内存不足，您可以尝试以下几件事：

1。）减少批量大小

2。）减小网络大小