How to fix RuntimeError: CUDA out of memory in Python | DevErrors