DeepInf bug: MemoryError: Unable to allocate 7.26 GiB for an array with shape (779164, 50, 50) and d

最新推荐文章于 2025-03-03 23:31:28 发布

原创最新推荐文章于 2025-03-03 23:31:28 发布 · 2.5k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python

在执行Python脚本train.py时遇到MemoryError，详细错误信息指出无法为形状(779164,50,50)的数组分配7.26GiB的空间，数据类型为float32。这通常发生在尝试加载或创建超过系统可用内存的大数组时。

Traceback (most recent call last):
  File "train.py", line 101, in <module>
    args.file_dir, args.dim, args.seed, args.shuffle, args.model)
  File "/root/DeepInf/data_loader.py", line 48, in __init__
    self.graphs = np.load(os.path.join(file_dir, "adjacency_matrix.npy")).astype(np.float32)
MemoryError: Unable to allocate 7.26 GiB for an array with shape (779164, 50, 50) and data type float32