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
- 内存一定是足够的,可能是内存分配碎片化严重,导致无法合理分配---->解决方法:重启电脑、重启整个pycharm,让内存重新分配
- 虚拟机的RAM分配不足,可能需要去掉自动分配内存,具体打开本机的
高级系统环境变量
,按如下所示: