问题:
在使用pytorch_geometric训练GCN网络时,运行出现错误:
AttributeError: module 'torch' has no attribute "'sparse_csr'"
出现以上问题,是因为下载的torch_geometric版本过高导致的,我的是2.3.1版本,太高了,需要降低版本。
方法:
1.
先激活虚拟环境,输入:
pip uninstall torch_geometric
卸载已安装的torch_geometric.
2.
指定torch_geometric版本安装:
pip install torch_geometric==2.0.4 -i https://pypi.tuna.tsinghua.edu.cn/simple
等待安装完成即可。
3.
想看更详细的pytorch_geometric这个API的安装步骤可以参考我的另一篇文章:MFBZS:torch_geometric详细安装教程
收藏不迷路~