Unable to allocate 910MiB for an array with shape (24000, 12, 207, 2) with type float64

910MiB = 24000 * 12 * 207 * 2 * 8 / 1024 / 1024

本次bug发生在Graph WaveNet代码中的dataloader['train_loader'].shuffle()部分。

分析:本机是16GB内存+10GB GPU

  1. 内存一定是足够的,可能是内存分配碎片化严重,导致无法合理分配---->解决方法:重启电脑、重启整个pycharm,让内存重新分配
  2. 虚拟机的RAM分配不足,可能需要去掉自动分配内存,具体打开本机的高级系统环境变量,按如下所示:在这里插入图片描述
    在这里插入图片描述
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐