PyTorch CUDA内存不足

时间:2020-08-30 18:23:06

标签: pytorch

我正在尝试使用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]**

我试图了解此错误表示什么?有什么建议可以解决吗?

0 个答案:

没有答案