torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 1226.19 GiB. GPU has a total capacity of 63.98 GiB of which 36.29 GiB is free. Of the allocated memory 24.03 GiB is allocated by ...