MemoryError: Unable to allocate 6.73 GiB for an array with shape (3000, 3, 448, 448) and data type

问题原因:内存过载,

现象描述:当时是使用TensorFlow2.0-gpu训练数据,当模型训练时批次过大,就会报以上错误。显卡是RTX 2060 6g 我当时就在想,我的显卡不至于这样弱吧。最后查阅了哈资料,发现是数据处理,占用的的CPU的RAM,如果你分配过少,就会报上面的错误哈。

如果大家的显卡比较好的话,出现了这种问题,那么大家可以重启一下,或者把一些暂用GPU的软件关了   因为发现GPU是先占用就先用,而导致后面需要使用GPU的软件 会因为分配不到足够的GPU内存而报错,但是如果你先跑程序 在打开其他软件  就不会报错  如果这个方法不行,大家再去操作下面的步骤。

解决方法:增加占用RAM的内存

1、点击设置,搜索控制面板,点击系统和安全

2、点击系统 ,在点击高级系统设置

3、点击高级->点击设置

4、 点击高级->点击更改

5、我这里输入的是6g,你可以输入自己的大小

7、点击确定即可

猜你喜欢

转载自blog.csdn.net/qq_40214464/article/details/118034670