今天在使用 pip 安装 torch-scatter 包时报错，然后使用 conda 成功安装了，但是运行代码的时候报错，并且是一个底层的错误。
查询之后发现，是因为在使用 torch_scatter、torch_sparse、torch_cluster 等基于 C++ 的扩展模块时，如果 PyTorch 版本和系统环境不完全匹配，就容易发生这类底层编译失败。
解决方法最稳定也最省事的方式是不要从源码编译 torch_scatter，而是使用预编译好的 .whl 包进行安装。步骤如下：
- 确认
PyTorch和Python版本
1 | python -c "import torch; print(torch.__version__)" |
- 访问 https://data.pyg.org/whl/ 下载对应版本的
*.whl到本地
- 使用
pip安装下载的whl包
1 | pip install xxx.whl |
这样问题就解决了，开心😄