出现的问题：torch_geometric报错

会出现为torch_sparse torch_scatter等的问题

最近又开始搞图神经网络方面的东西，要用到torch_geometric。但是写好代码后发现会出现报错：

Traceback (most recent call last):
  File "d:/env/CProjects/GCN/GAT.py", line 10, in <module>
  File "D:\anaconda3\lib\site-packages\torch_geometric\__init__.py", line 4, in <module>
    import torch_geometric.data
  File "D:\anaconda3\lib\site-packages\torch_geometric\data\__init__.py", line 1, in <module>
    from .data import Data
  File "D:\anaconda3\lib\site-packages\torch_geometric\data\data.py", line 3, in <module>
    from torch_geometric.typing import OptTensor, NodeType, EdgeType
  File "D:\anaconda3\lib\site-packages\torch_geometric\typing.py", line 4, in <module>
    from torch_sparse import SparseTensor
ModuleNotFoundError: No module named 'torch_sparse'

不难看出是torch_geometric中调用torch_sparse时出现了问题，而且好像是没有安装，很自然的安装这些依赖包。

如何安装依赖包

可以说这些包是真的难装，我一直会出现ERROR: Failed building wheel for torch-sparse这个问题，按我的理解来讲，这个问题有时候极有可能是包之间的版本问题引起的。然后在网上查了一下，果然是这样的。

解决版本问题

1、卸载

先把torch-geometric、torch-sparse、torch-scatter、torch-cluster、 torch-spline-conv全部卸载了
pip uninstall torch-geometric torch-scatter torch-sparse torch-cluster torch-spline-conv

2、确定版本

conda list确定PyTorch的版本，我的是1.10
nvcc -V确定cuda的版本，我的是11.3

3、确定下载地址

CPU版本的地址如下：torch-1.10.0+cpu的地址
https://pytorch-geometric.com/whl/torch-1.10.0+cpu.html
GPU版本的地址如下：torch-1.10.0+cuda11.3的地址
https://pytorch-geometric.com/whl/torch-1.10.0+cu113.html

需要根据自己的版本更改网址！！！
网址能打开说明就ok，如果不存在那说明版本冲突

4、安装

我用是pip，通过-f指定链接

pip install torch-scatter torch-sparse torch-cluster torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.10.0+cu113.html

pip install torch-geometric

大功告成

标签：py,sparse,1.10,torch,报错,版本,geometric
来源： https://blog.csdn.net/qq_54478153/article/details/122683064