PyTorch CUDA内存不足

我正在尝试使用pytorch运行建模脚本。我尝试了批处理，一次只向我的模型发送30个视频

 File "/home/ubuntu/anaconda3/lib/python3.7/site-packages/torch/nn/modules/activation.py", line 988, in forward
    return F.prelu(input, self.weight)
  File "/home/ubuntu/anaconda3/lib/python3.7/site-packages/torch/nn/functional.py", line 1319, in prelu
    return torch.prelu(input, weight)
**RuntimeError: CUDA out of memory. Tried to allocate 2.19 GiB (GPU 0; 11.17 GiB total capacity; 7.43 GiB already allocated; 535.31 MiB free; 10.58 GiB reserved in total by PyTorch)
^M 83%|████████▎ | 5/6 [00:39<00:07,  7.82s/it]**

文件“/home/ubuntu/anaconda3/lib/python3.7/site packages/torch/nn/modules/activation.py”，第988行，向前
返回F.prelu（输入，自重）
文件“/home/ubuntu/anaconda3/lib/python3.7/site packages/torch/nn/functional.py”，第1319行，prelu
返回火炬预润滑（输入，重量）
**运行时错误：CUDA内存不足。尝试分配2.19 GiB（GPU 0；11.17 GiB总容量；7.43 GiB已分配；535.31 MiB空闲；PyTorch总共保留10.58 GiB）
^M 83%|████████▎ | 5/6[00:39错误表明它需要更多的GPU内存=]减少到最小批量大小=1，然后再增加。此外，您还可以为多个步骤（例如
32
）累积梯度，并对其执行单个优化器步骤，以稳定损失。