在导入 torch_geometric 抛出“std::bad_alloc”实例后终止调用

Question

我正在用 python 编写并收到错误：

“抛出'std :: bad_alloc'实例后调用终止。what
（）：std :: bad_alloc。
中止（核心转储）”

经过大量的调试，我发现问题的根源是：

import torch_geometric

我什至只用这行代码创建了一个文件，但仍然收到错误。
我正在 conda 环境（4.10.3）中运行，我确保在 conda 环境中安装了 torch_geometric。我尝试删除并重新安装，但这不起作用。
我还尝试删除并重新安装 torch/cuda。
我用谷歌搜索了这个错误，但似乎只提出了数据分配方面的问题，但我不确定这会是一个问题，因为我只是导入 torch_geometric。

有任何想法吗？

Answer 1

这个问题是因为pytorch版本不匹配造成的。当前使用的pytorch是1.11.0，但是当安装了scatter和sparse时安装了scatter和sparse，使用了1.10.1：

• pip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.1+cu113.html。
• pip install torch-sparse -f https://data.pyg.org/whl/torch-1.10.1+cu113.html

因此，torch-1.10.1用于安装scatter和sparse，但torch-1.11.0才是真正的版本。

简单地做：

• pip 卸载 torch-scatter
• pip 卸载 torch-sparse
• pip install torch-scatter -f https://data.pyg.org/whl/torch-1.11.0+cu113.html。
• pip install torch-sparse -f https://data.pyg.org/whl/torch-1.11.0+cu113.html

解决问题。