Hi Lara, you ran out of memory. Unfortunately.

Try out the suggested solution.

If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF