torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 20.00 MiB. GPU

torch.cuda.OutOfMemoryError是由于GPU内存不足而引发的错误。当你的程序在使用GPU时，如果请求的内存超过了GPU的可用内存，则会发生这个错误。这通常是由于你的模型或数据集过大，或者你的计算机没有足够的显存来完成所需的操作。

为了解决这个问题，你可以尝试以下几种方法：
1. 减少batch size：减少每个batch的样本数量可以降低GPU的内存使用量。
2. 减少模型的大小：通过减少模型参数的数量、使用更小的网络结构等方式减少模型的大小，从而减少GPU内存的使用。
3. 优化代码：使用PyTorch提供的内存优化工具、减少无用计算等方式优化代码，减少GPU内存的使用。
4. 使用更大显卡：如果你的计算机显卡比较老旧，可能显存较小，考虑更换更大显存的显卡。