RuntimeError: CUDA out of memory.

来源：4-7 利用神经网络解决分类和回归问题（5）

weixin_慕尼黑0465866

2021-09-26

RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 2.00 GiB total capacity; 1.21 GiB already allocated; 16.13 MiB free; 19.62 MiB cached)
图片描述
遇到这种情况，是不是只能换显卡了呀？

写回答

1回答

会写代码的好厨师

2021-10-11

显存太小了。解决办法：1. 减少训练的batchsize；2.裁剪模型尺寸。3；换一块显卡

PyTorch入门到进阶实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发，快速掌握PyTorch框架

1190 学习 · 293 问题

查看课程

相似问题

mmdetection训练VOC数据集时遇到RuntimeError: CUDA out of memory.

回答 1

cuda安装

回答 1

google colab pro的cuda跑的很慢怎么办？

回答 1

cuda安装问题

回答 1

这里用windows，没有cuda可以吗

回答 1

打开慕课网App查看更多内容