RuntimeError: CUDA error: out of memory code example
Example: RuntimeError: CUDA out of memory.
Your GPU is out of memory,
reduce your batch size until your code runs without this error
# also, find any processes using gpu and kill them with:
nvidia-smi
kill -9 [pid]