PyGOD 使用 成功解决 Runtimeerror: No such operator torch_sparse::hetero_neighbor_sample 报错

Posted 叶庭云

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了PyGOD 使用 成功解决 Runtimeerror: No such operator torch_sparse::hetero_neighbor_sample 报错相关的知识,希望对你有一定的参考价值。

问题描述:

按照教程一步步装好了 PyGOD,用到的命令如下所示(我的环境:PyTorch1.8.1 + CUDA11.1):

pip list
nvidia-smi
nvcc --version


pip install networkx
pip install scikit-learn==0.23.1


pip install torch-scatter==2.0.8 -f https://pytorch-geometric.com/whl/torch-1.8.1+cu111.html
pip install torch-cluster==1.5.9 -f https://pytorch-geometric.com/whl/torch-1.8.1+cu111.html
pip install torch-sparse==0.6.10 -f https://pytorch-geometric.com/whl/torch-1.8.1+cu111.html
pip install torch-spline-conv==1.2.1 -f https://pytorch-geometric.com/whl/torch-1.8.1+cu111.html
pip install torch-geometric -f https://pytorch-geometric.com/whl/torch-1.8.1+cu111.html

pip install --upgrade pygod -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

然而,运行代码测试图异常检测时,报错如下:

from pygod.utils import load_data

data = load_data('inj_cora')
data.y = data.y.bool()

from pygod.models import DOMINANT

model = DOMINANT()

model.fit(data)
......
Runtimeerror: No such operator torch_sparse::hetero_neighbor_sample
......

分析觉得是 torch_sparse 版本的问题,卸载后重新安装 0.6.12 版本解决了问题:

pip uninstall torch-sparse
pip install torch-sparse==0.6.12 -f https://pytorch-geometric.com/whl/torch-1.8.1+cu111.html

References:

以上是关于PyGOD 使用 成功解决 Runtimeerror: No such operator torch_sparse::hetero_neighbor_sample 报错的主要内容,如果未能解决你的问题,请参考以下文章

成功自动解决合并冲突后,使用 KDiff 审查合并操作

使用npm安装electron不成功的解决方法

使用mysqldump导入不成功解决方法

使用 Spring 的 AbstractAuthenticationProcessingFilter 成功认证后解决 404

解决PPTP客户端拨号不成功

使用 Elastic 8.4 搜索解决问题并取得成功