Как решить этот вопрос "RuntimeError: CUDA out of memory."?

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

Поскольку вашему графическому процессору не хватает памяти, вы можете попробовать несколько вещей:

1.) Уменьшите размер партии

2.) Уменьшите размер сети