RuntimeError: CUDA out of memory. Tried to allocate 192.00 MiB (GPU 0; 12.00 GiB total capacity; 1.66 GiB already allocated; 7.97 GiB free; 1.82 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

这个错误是由于 CUDA 内存溢出导致的。它告诉你在 GPU 0 上尝试分配 192.00 MiB 的内存，但是已经有 1.66 GiB 的内存被分配了，剩余 7.97 GiB 的空闲内存不足以满足你的需求。这时你可以尝试设置 PyTorch 的内存管理参数，比如 `max_split_size_mb`，以避免碎片化。你可以查阅 PyTorch 的文档了解更多关于内存管理和 `PYTORCH_CUDA_ALLOC_CONF` 的信息。