翻译成中文：torchcudaOutOfMemoryError CUDA out of memory Tried to allocate 1600 MiB GPU 0; 400 GiB total capacity; 300 GiB already allocated; 0 bytes free; 301 GiB reserved in total by PyTorch If reserved m

torch.cuda.OutOfMemoryError: CUDA内存不足。尝试分配16.00 MiB（GPU 0；总容量4.00 GiB；已分配3.00 GiB；无空闲字节；PyTorch总共保留了3.01 GiB）。如果保留的内存>>已分配的内存，请尝试设置max_split_size_mb以避免碎片化。有关内存管理和PYTORCH_CUDA_ALLOC_CONF，请参阅文档。