Large datasets and Cuda memory Issue

I was processing a large dataset and ran into this error: "RuntimeError: CUDA out of memory. Tried to allocate 1.35 GiB (GPU 0; 8.00 GiB total capacity; 3.45 GiB already allocated; 1.20 GiB free; 4.79 GiB reserved in total by PyTorch).

Any thought on how to solve this?

https://stackoverflow.com/questions/66997068/large-datasets-and-cuda-memory-issue April 08, 2021 at 11:47AM