我正在尝试使用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]**
我试图了解此错误表示什么?有什么建议可以解决吗?