[Solved][PyTorch] RuntimeError: CUDA out of memory. Tried to allocate 2.0 GiB
Today I want to record a common problem, its solution is very rarely. Simple to put, the error message as follow:
RuntimeError: CUDA out of memory. Tried to allocate 2.0 GiB.