RuntimeError: CUDA out of memory. Tried to allocate 1.16 GiB (GPU 0; 3.81 GiB total capacity; 1.06 GiB already allocated; 1.17 GiB free; 1.10 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

这个错误是由于您的GPU内存不足导致的。您可以尝试以下解决方法：

1. 减少模型批处理大小（batch size），这样可以减少内存占用。
2. 使用更小的模型，或者尝试使用更轻量级的预训练模型。
3. 如果您的机器有多个GPU，可以尝试将模型加载到其他GPU上。
4. 清理GPU内存。您可以使用`torch.cuda.empty_cache()`来清理已分配的缓存。
5. 如果您使用的是PyTorch 1.6或更高版本，可以尝试设置`max_split_size_mb`参数来避免内存碎片化。具体设置方法请查阅PyTorch的文档。

希望这些方法能够帮助您解决CUDA内存不足的问题！