微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

PyTorch CUDA内存不足

如何解决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]**

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

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。