It does not complete the first epoch after several hours of training. And at one point, the CUDA Out of Memory error message prompt:
RuntimeError: CUDA out of memory. Tried to allocate 12.50 MiB (GPU 0; 10.92 GiB total capacity; 8.57 MiB already allocated; 9.28 GiB free; 4.68 MiB cached is displayed.