运行此项目所需的 GPU 内存是多少?

torch.cuda.OutOfMemoryError: CUDA 内存不足。尝试分配 2.17 GiB 内存（GPU 0；总容量 14.84 GiB；已分配 11.81 GiB；可用 1.67 GiB；PyTorch 总共预留 12.22 GiB 内存）。如果预留的内存 > 分配的内存，请尝试设置 max_split_size_mb 以避免碎片化。请参阅文档中的内存管理和 PYTORCH_CUDA_ALLOC_CONF 相关信息。