Is not working for me:
OOM: Ran out of memory with exception: CUDA out of memory.
Tried to allocate 5.59 GiB (GPU 0; 10.92 GiB total capacity; 4.28 GiB
already allocated; 5.14 GiB free; 5.20 GiB reserved in total by PyTorch)