CUDA out of memory in image denoising code.

[MayThetTun]

restored = model_restoration(input_) in denoising.
RuntimeError: CUDA out of memory. Tried to allocate 176.00 MiB (GPU 0; 5.00 GiB total capacity; 3.94 GiB already allocated; 0 bytes free; 4.28 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF .

[chenguo12315]

May be your image's is not a eight multiple you should resize your image size

[hust-lidelong]

已收到邮件，谢谢

[ahmedAbassi123]

what batch size are you using?