Solution for Runtime Error: CUDA out of memory

"RuntimeError: CUDA out of memory. ~" in PyTorch (Windows) with CUDA 11.0 & NVIDIA RTX 3090

This error is related to the number of workers.

To solve this issue, reduce the number of workers in DataLoader

 

# Reference:

discuss.pytorch.org/t/cuda-out-of-memory-while-increase-num-workers-in-dataloader/101877