Same here…
......
RuntimeError: CUDA out of memory. Tried to allocate 384.00 MiB (GPU 0; 10.76 GiB total capacity; 8.84 GiB already allocated; 117.62 MiB free; 8.90 GiB reserved in total by PyTorch)

What a PyTorch … Extremely sad…