解决：RuntimeError: CUDA out of memory. Tried to allocate 2.00 MiB

素颜马尾好姑娘i 2023-10-04 17:01 76阅读 0赞

## **方法一：** ##

仅需**减小batchsize**

改文件的配置cfg的**batchsize=4**，一般在cfg文件下的查找batch或batchsize，将batchsize调小后，再次运行，类似于改下面

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTUyOTA5Mw_size_16_color_FFFFFF_t_70]

## 方法二 ： ##

上述方法还没解决，不改batchsize，可以考虑下面的方法的链接

不计算梯度：

**ps: 在报错的哪一行代码的上面，加上下面一行代码，不计算梯度**

with torch.no\_grad()

[不计算梯度的方法][Link 1]

## 方法三： ##

释放内存：链接如下

[释放内存][Link 2]

if hasattr(torch.cuda, 'empty_cache'):
    	torch.cuda.empty_cache()

**ps: 在报错的哪一行代码的上面，加上下面两行代码，释放无关的内存**

if hasattr(torch.cuda, 'empty_cache'):
    	torch.cuda.empty_cache()

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTUyOTA5Mw_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20210314111519148.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTUyOTA5Mw==,size_16,color_FFFFFF,t_70
[Link 1]: https://blog.csdn.net/qq_34907927/article/details/109909133?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.control&dist_request_id=1328642.36354.16156873028290129&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.control
[Link 2]: https://blog.csdn.net/weixin_32393347/article/details/104395179